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

remove lr_dropout for plaidml backend

Browse source
This commit is contained in:
Colombo 2020-01-08 11:11:33 +04:00
parent d3e6b435aa
commit b8182ae42b
2 changed files with 9 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

5
models/Model_Quick96/Model.py
View file

@ -141,8 +141,9 @@ class Quick96Model(ModelBase):
self.CA_conv_weights_list += [layer.weights[0]] #- is Conv2D kernel_weights
if self.is_training_mode:
self.src_dst_opt = RMSprop(lr=2e-4, lr_dropout=0.3)
self.src_dst_mask_opt = RMSprop(lr=2e-4, lr_dropout=0.3)
lr_dropout = 0.3 if nnlib.device.backend != 'plaidML' else 0.0
self.src_dst_opt = RMSprop(lr=2e-4, lr_dropout=lr_dropout)
self.src_dst_mask_opt = RMSprop(lr=2e-4, lr_dropout=lr_dropout)
target_src_masked = self.model.target_src*self.model.target_srcm
target_dst_masked = self.model.target_dst*self.model.target_dstm