github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-06 04:52:13 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

SAE: revert back to sigmoid. Trainings must be restarted.

Browse source
This commit is contained in:
iperov 2019-03-16 09:01:11 +04:00
parent ea5fe28d08
commit 5076feb90f
2 changed files with 21 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

3
models/ModelBase.py
View file

@ -410,6 +410,9 @@ class ModelBase(object):
def get_training_data_generators (self):
return self.generator_list
def get_model_root_path(self):
return self.model_path
def get_strpath_storage_for_file(self, filename):
if self.device_args['force_gpu_idx'] == -1:
return str( self.model_path / ( self.get_model_name() + '_' + filename) )