updated loss in inverse render

marcoamonteiro · Jun 30, 2021 · 0800af7 · 0800af7
1 parent e862696
commit 0800af7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/inverse_render.py b/inverse_render.py
@@ -106,7 +106,7 @@ def tensor_to_PIL(img):
     noise_w_frequencies = 0.03 * torch.randn_like(w_frequencies) * (n_iterations - i)/n_iterations
     noise_w_phase_shifts = 0.03 * torch.randn_like(w_phase_shifts) * (n_iterations - i)/n_iterations
     frame, _ = generator.forward_with_frequencies(w_frequencies + noise_w_frequencies + w_frequency_offsets, w_phase_shifts + noise_w_phase_shifts + w_phase_shift_offsets, **options)
-    loss = torch.nn.L1Loss()(frame, gt_image)
+    loss = torch.nn.MSELoss()(frame, gt_image)
     loss = loss.mean()