Update train_simple.py

MingtaoGuo · Oct 12, 2022 · 2004654 · 2004654
1 parent 9e2ad0e
commit 2004654
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train_simple.py b/train_simple.py
@@ -205,7 +205,7 @@ def train(args, loader, generator, discriminator, arcface, vgg19, g_optim, d_opt
         bce_loss = F.binary_cross_entropy(fake_mask, mask).mean()
         # bce_loss = F.l1_loss(fake_mask, mask).mean()
         # ----------- total loss -------------
-        g_loss = adv_loss + 10 * id_loss + bce_loss + 100 * fm_loss + 100 * rec_loss
+        g_loss = adv_loss + 10 * id_loss + bce_loss + 10 * fm_loss + 10 * rec_loss
         loss_dict['g'] = g_loss
         loss_dict['adv'] = adv_loss
         loss_dict['id'] = id_loss