github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-08-22 14:24:40 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

bug fix modifier

Browse source
This commit is contained in:
Jan 2021-12-09 16:54:26 +01:00
commit 447ac28b4d
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
models/Model_SAEHD/Model.py
View file

@ -370,7 +370,7 @@ class SAEHDModel(ModelBase):
lr_modifier = self.options['lr_modifier'] lr_modifier = self.options['lr_modifier']
if lr_modifier == 0: if lr_modifier == 0:
lr = 5e-5 lr = 5e-5
elif lr_modifier > 0: elif lr_modifier < 0:
lr = 5e-5 / abs( lr_modifier * 4/100 ) lr = 5e-5 / abs( lr_modifier * 4/100 )
else: else:
lr = 5e-5 * abs( lr_modifier * 4/100 ) lr = 5e-5 * abs( lr_modifier * 4/100 )