feat: giving option that percep can be zero for Danbooru percp

Kiteretsu77 · Mar 28, 2024 · eb926e0 · eb926e0
1 parent 7d67eb1
commit eb926e0
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/train_code/train_grlgan.py b/train_code/train_grlgan.py
@@ -116,9 +116,12 @@ def calculate_loss(self, gen_hr, imgs_hr):
 
 
         # Generator perceptual loss:        generated vs. perceptual
-        l_g_percep_danbooru = self.cri_danbooru_perceptual(gen_hr, imgs_hr)
         l_g_percep_vgg = self.cri_vgg_perceptual(gen_hr, imgs_hr)
-        l_g_percep = l_g_percep_danbooru + l_g_percep_vgg 
+        l_g_percep = l_g_percep_vgg 
+        # For the optional Anime Perceputla Loss
+        if self.options["danbooru_perceptual_loss_weight"] != 0.0:
+            l_g_percep_danbooru = self.cri_danbooru_perceptual(gen_hr, imgs_hr)
+            l_g_percep += l_g_percep_danbooru
         self.generator_loss += l_g_percep
         self.weight_store["perceptual_loss"] = l_g_percep