From 9137fa94bf2957cc4f4a64874f174737751bd642 Mon Sep 17 00:00:00 2001
From: Solal Nathan <otthorn@crans.org>
Date: Fri, 30 Apr 2021 17:04:27 +0200
Subject: [PATCH] Add tqdm to the main optimisation loop

---
 main.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/main.py b/main.py
index 97c124e..8e55562 100644
--- a/main.py
+++ b/main.py
@@ -523,7 +523,7 @@ n_iter = 3000
 loss_history_color, loss_history_sil = [], []
 
 # The main optimization loop.
-for iteration in range(n_iter):
+for iteration in tqdm(range(n_iter)):
     # In case we reached the last 75% of iterations,
     # decrease the learning rate of the optimizer 10-fold.
     if iteration == round(n_iter * 0.75):