Add tqdm to the main optimisation loop

cli
otthorn 3 years ago
parent 6b1829a893
commit 9137fa94bf

@ -523,7 +523,7 @@ n_iter = 3000
loss_history_color, loss_history_sil = [], [] loss_history_color, loss_history_sil = [], []
# The main optimization loop. # 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, # In case we reached the last 75% of iterations,
# decrease the learning rate of the optimizer 10-fold. # decrease the learning rate of the optimizer 10-fold.
if iteration == round(n_iter * 0.75): if iteration == round(n_iter * 0.75):

Loading…
Cancel
Save