simcop2387/faceswap
1
0
Fork 0
mirror of https://github.com/deepfakes/faceswap synced 2025-06-07 10:43:27 -04:00
Code Issues Releases Wiki Activity

Bugfix: Handle distribution strat in convert

Browse source
This commit is contained in:
torzdf 2022-07-20 12:44:22 +01:00
parent 54398559ff
commit dcb436c9df
1 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
plugins/train/model/_base/settings.py
View file

@ -405,7 +405,11 @@ class Settings():
if self._use_mixed_precision:
logger.info("Enabling Mixed Precision Training.")
self._strategy = self._get_strategy(arguments.distribution_strategy)
if hasattr(arguments, "distribution_strategy"):
strategy = arguments.distribution_strategy
else:
strategy = "default"
self._strategy = self._get_strategy(strategy)
logger.debug("Initialized %s", self.__class__.__name__)
@property