fix bug header of valid.csv

dingguanglei · Aug 9, 2019 · a69150b · a69150b
1 parent bf4f141
commit a69150b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/jdit/trainer/gan/sup_gan.py b/jdit/trainer/gan/sup_gan.py
@@ -220,7 +220,7 @@ def valid_epoch(self):
             avg_dic[key] = avg_dic[key] / self.datasets.nsteps_valid
 
         self.watcher.scalars(avg_dic, self.step, tag="Valid")
-        self.loger.write(self.step, self.current_epoch, avg_dic, "Valid", header=self.step <= 1)
+        self.loger.write(self.step, self.current_epoch, avg_dic, "Valid", header=self.current_epoch <= 1)
         self._watch_images(tag="Valid")
         self.netG.train()
         self.netD.train()

diff --git a/jdit/trainer/single/sup_single.py b/jdit/trainer/single/sup_single.py
@@ -189,7 +189,7 @@ def valid_epoch(self):
             avg_dic[key] = avg_dic[key] / self.datasets.nsteps_valid
 
         self.watcher.scalars(avg_dic, self.step, tag="Valid")
-        self.loger.write(self.step, self.current_epoch, avg_dic, "Valid", header=self.step <= 1)
+        self.loger.write(self.step, self.current_epoch, avg_dic, "Valid", header=self.current_epoch <= 1)
         self._watch_images(tag="Valid")
         self.net.train()