From fdc41cd21273ad6de0a269dd5d0a0ced5d6eb059 Mon Sep 17 00:00:00 2001 From: jh Date: Thu, 11 Mar 2021 23:38:08 -0800 Subject: [PATCH] fix: model path --- mainscripts/Trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mainscripts/Trainer.py b/mainscripts/Trainer.py index dabec6d..fd55f84 100644 --- a/mainscripts/Trainer.py +++ b/mainscripts/Trainer.py @@ -405,7 +405,7 @@ def main(**kwargs): batch_size, zoom) # io.show_image(wnd_name, preview_pane_image) - model_path = Path(kwargs.get('model_path', '')) + model_path = Path(kwargs.get('saved_models_path', '')) filename = 'preview.jpg' preview_file = str(model_path / filename) cv2.imwrite(preview_file, preview_pane_image)