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

no message

Browse source
This commit is contained in:
jh 2021-03-12 18:32:49 -08:00
commit 39601603ef
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
core/leras/layers/MsSsim.py
View file

@ -22,7 +22,7 @@ class MsSsim(nn.LayerBase):
def assign_device(op):
if op.type != 'Assert':
if op.type != 'Assert' or op.type != 'ListDiff':
return '/gpu:0'
else:
return '/cpu:0'