SupGanTrain refactor.

dingguanglei · Nov 9, 2018 · a704c53 · a704c53
1 parent 6159088
commit a704c53
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 2 deletions.
diff --git a/.gitignore b/.gitignore
@@ -3,6 +3,7 @@
 #mypackage/metric/Cifar10_M.csv
 #mypackage/metric/Cifar10_S.csv
 
+log
 jdit/data
 .idea
 datasets

diff --git a/jdit/trainer/instances/cifarPix2pixGan.py b/jdit/trainer/instances/cifarPix2pixGan.py
@@ -108,7 +108,7 @@ class CifarPix2pixGanTrainer(Pix2pixGanTrainer):
     def __init__(self, logdir, nepochs, gpu_ids_abs, netG, netD, optG, optD, datasets):
         super(CifarPix2pixGanTrainer, self).__init__(logdir, nepochs, gpu_ids_abs, netG, netD, optG, optD,
                                                      datasets)
-        self.datasets.dataset_train = self.datasets.dataset_valid
+
         self.watcher.graph(netG, (4, 1, 32, 32), self.use_gpu)
 
     def get_data_from_loader(self, batch_data):

diff --git a/jdit/trainer/instances/fashingClassification.py b/jdit/trainer/instances/fashingClassification.py
@@ -35,7 +35,6 @@ class FashingClassTrainer(ClassificationTrainer):
 
     def __init__(self, logdir, nepochs, gpu_ids, net, opt, datasets):
         super(FashingClassTrainer, self).__init__(logdir, nepochs, gpu_ids, net, opt, datasets)
-
         self.watcher.graph(net, self.datasets.batch_shape, self.use_gpu)
         data, label = self.datasets.samples_train
         self.watcher.embedding(data, data, label)