OSSForks/faceswap
1
0
Fork 0
mirror of https://github.com/zebrajr/faceswap.git synced 2026-01-15 12:15:15 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

bugfix: Catch learn-mask error when no mask type selected

Browse Source
This commit is contained in:
torzdf
2022-10-12 12:03:59 +01:00
parent f1e3339fbe
commit 403f981e0f
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
plugins/train/model/_base/model.py
View File

@@ -129,6 +129,10 @@ class ModelBase():
"Mask to use. Please select a mask or disable Penalized Mask "
"Loss.")
if self.config["learn_mask"] and self.config["mask_type"] is None:
raise FaceswapError("'Learn Mask' has been selected but you have not chosen a Mask to "
"use. Please select a mask or disable 'Learn Mask'.")
self._mixed_precision = self.config["mixed_precision"] and get_backend() != "amd"
# self._io = IO(self, model_dir, self._is_predict, self.config["save_optimizer"])
# TODO - Re-enable saving of optimizer once this bug is fixed: