github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-08-22 06:23:20 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: create_preview_pane_image

Browse source
This commit is contained in:
jh 2021-03-11 23:35:43 -08:00
commit 2320fed26b
1 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
mainscripts/Trainer.py
View file

@ -308,7 +308,9 @@ def create_preview_pane_image(previews, selected_preview, loss_history,
else:
loss_history_to_show = loss_history[-show_last_history_iters_count:]
lh_height = int(100 * zoom.scale)
lh_img = models.ModelBase.get_loss_history_preview(loss_history_to_show, iteration, batch_size, w, c, lh_height)
# FIXME: investigate old behavior (batch_size, lh_height)
# lh_img = models.ModelBase.get_loss_history_preview(loss_history_to_show, iteration, batch_size, w, c, lh_height)
lh_img = models.ModelBase.get_loss_history_preview(loss_history_to_show, iteration, w, c)
final = np.concatenate([final, lh_img], axis=0)
final = np.concatenate([final, selected_preview_rgb], axis=0)