From 2320fed26bb05284c36bab22df008d5f6e12bf25 Mon Sep 17 00:00:00 2001 From: jh Date: Thu, 11 Mar 2021 23:35:43 -0800 Subject: [PATCH] fix: create_preview_pane_image --- mainscripts/Trainer.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/mainscripts/Trainer.py b/mainscripts/Trainer.py index 0e81831..dabec6d 100644 --- a/mainscripts/Trainer.py +++ b/mainscripts/Trainer.py @@ -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)