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: model path

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

2
mainscripts/Trainer.py
View file

@ -405,7 +405,7 @@ def main(**kwargs):
batch_size, batch_size,
zoom) zoom)
# io.show_image(wnd_name, preview_pane_image) # 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' filename = 'preview.jpg'
preview_file = str(model_path / filename) preview_file = str(model_path / filename)
cv2.imwrite(preview_file, preview_pane_image) cv2.imwrite(preview_file, preview_pane_image)