mirror of
https://github.com/iperov/DeepFaceLab.git
synced 2025-07-16 10:03:41 -07:00
fix mean loss history after save
This commit is contained in:
parent
9fd49ee3f0
commit
3e63ba2d17
1 changed files with 1 additions and 1 deletions
|
@ -133,7 +133,7 @@ def trainerThread (s2c, c2s, e,
|
|||
shared_state['after_save'] = False
|
||||
last_save_time = time.time()
|
||||
|
||||
mean_loss = np.mean ( [ np.array(loss_history[i]) for i in range(save_iter, iter) ], axis=0)
|
||||
mean_loss = np.mean ( loss_history[save_iter:iter], axis=0)
|
||||
|
||||
for loss_value in mean_loss:
|
||||
loss_string += "[%.4f]" % (loss_value)
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue