Minor bug fixes

Whoo-jl · Sep 18, 2020 · f06cce4 · f06cce4
1 parent 5265ef2
commit f06cce4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/config/super_resolution.yaml b/src/config/super_resolution.yaml
@@ -8,7 +8,7 @@ TRAIN:
   ]
   batch_size: 512
   width: 128
-  height: 32
+  height: 64
   epochs: 500
   cuda: True
   ngpu: 4

diff --git a/src/interfaces/base.py b/src/interfaces/base.py
@@ -149,7 +149,7 @@ def generator_init(self):
         if self.args.arch != 'bicubic':
             model = model.to(self.device)
             image_crit.to(self.device)
-            if cfg.ngpu > 1:
+            if cfg.ngpu >= 1:
                 model = torch.nn.DataParallel(model, device_ids=range(cfg.ngpu))
                 image_crit = torch.nn.DataParallel(image_crit, device_ids=range(cfg.ngpu))
             if self.resume is not '':