This commit is contained in:
Nataniel Ruiz
2019-12-24 12:51:35 -04:00
parent 3f52c828a7
commit f665283dfa
+2 -2
View File
@@ -635,8 +635,8 @@ class Solver(object):
l1_error += F.l1_loss(gen, gen_noattack)
l2_error += F.mse_loss(gen, gen_noattack)
l0_error += np.linalg.norm((gen - gen_noattack).data.cpu(), ord=0)
min_dist += attacks.min_dist((gen - gen_noattack).data.cpu(), ord='-inf')
l0_error += np.linalg.norm((gen - gen_noattack).data[0].cpu(), ord=0)
min_dist += attacks.min_dist((gen - gen_noattack).data[0].cpu(), ord='-inf')
n_samples += 1
# Save the translated images.