reqst

dingguanglei · Oct 26, 2018 · 89f3b72 · 89f3b72
1 parent 164fbd1
commit 89f3b72
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/jdit/trainer/gan/generate.py b/jdit/trainer/gan/generate.py
@@ -3,7 +3,7 @@
 from tqdm import tqdm
 from torch.autograd import Variable
 # from jdit.metric.inception import FID
-# from ...metric import FID
+from ...metric import FID
 import torch
 
 
@@ -31,7 +31,7 @@ def __init__(self, logdir, nepochs, gpu_ids_abs, netG, netD, optG, optD, dataset
         self.fake = None
         self.fixed_input = None
         self.latent_shape = latent_shape
-        # self.metric = FID(self.gpu_ids)
+        self.metric = FID(self.gpu_ids)
         self.loger.regist_config(self.netG, config_filename="Generator")
         self.loger.regist_config(self.netD, config_filename="Discriminator")
         self.loger.regist_config(datasets)