CalvinBackup/disrupting-deepfakes
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

next

Browse Source
This commit is contained in:
Nataniel Ruiz
2019-12-25 20:37:23 -04:00
parent 1102475fe3
commit f93c2aec3e
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
cyclegan/models/test_model.py
View File

@@ -94,6 +94,10 @@ class TestModel(BaseModel):
l2 = F.mse_loss(generated, generated_noattack)
l0 = (generated - generated_noattack).norm(0)
d = (generated - generated_noattack).norm(float('-inf'))
generated = None
generated_noattack = None
return l1, l2, l0, d
def optimize_parameters(self):