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 committed 2019-12-25 20:37:23 -04:00
1 parent 1102475fe3
commit f93c2aec3e
1 file changed
+4
Download Patch File Download Diff File Expand all files Collapse all files
cyclegan/models/test_model.py
+4
View File
@@ -94,6 +94,10 @@ class TestModel(BaseModel):
l2 = F.mse_loss(generated, generated_noattack) l2 = F.mse_loss(generated, generated_noattack)
l0 = (generated - generated_noattack).norm(0) l0 = (generated - generated_noattack).norm(0)
d = (generated - generated_noattack).norm(float('-inf')) d = (generated - generated_noattack).norm(float('-inf'))
generated = None
generated_noattack = None
return l1, l2, l0, d return l1, l2, l0, d
def optimize_parameters(self): def optimize_parameters(self):