otthorn/nerf_tutorial
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Add tqdm to the main optimisation loop

Browse source
This commit is contained in:
otthorn 2021-04-30 17:04:27 +02:00
parent 6b1829a893
commit 9137fa94bf
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
main.py
View file

@ -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):