diff --git a/stargan/solver.py b/stargan/solver.py index c85162c..60bad1e 100644 --- a/stargan/solver.py +++ b/stargan/solver.py @@ -639,8 +639,6 @@ class Solver(object): l0_error += (gen - gen_noattack).norm(0) min_dist += (gen - gen_noattack).norm(float('-inf')) n_samples += 1 - - break # Save the translated images. x_concat = torch.cat(x_fake_list, dim=3)