Add tqdm to the main optimisation loop
This commit is contained in:
parent
6b1829a893
commit
9137fa94bf
1 changed files with 1 additions and 1 deletions
2
main.py
2
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):
|
||||
|
|
Loading…
Reference in a new issue