From b2d887e224347e83d3bf2412f100eb0c7a34c802 Mon Sep 17 00:00:00 2001
From: "Thomas \"elfprince13\" Dickerson" <elfprince13@gmail.com>
Date: Fri, 4 May 2018 15:13:59 -0400
Subject: More robust disabling of disabled losses.

---
 train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'train.py')

diff --git a/train.py b/train.py
index 231662c..368dc9f 100755
--- a/train.py
+++ b/train.py
@@ -61,7 +61,7 @@ for epoch in range(start_epoch, opt.niter + opt.niter_decay + 1):
 
         # calculate final loss scalar
         loss_D = (loss_dict['D_fake'] + loss_dict['D_real']) * 0.5
-        loss_G = loss_dict['G_GAN'] + loss_dict['G_GAN_Feat'] + loss_dict['G_VGG']
+        loss_G = loss_dict['G_GAN'] + loss_dict.get('G_GAN_Feat',0) + loss_dict.get('G_VGG',0)
 
         ############### Backward Pass ####################
         # update generator weights
-- 
cgit v1.2.3-70-g09d2