fix some bugs

myeldib · Jan 16, 2020 · 6c130aa · 6c130aa
1 parent a6994b5
commit 6c130aa
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/models/video_model.py b/models/video_model.py
@@ -8,11 +8,11 @@
 if Norm == 'instance':
     NormLayer_2d = nn.InstanceNorm2d
     NormLayer_3d = nn.InstanceNorm3d
-    use_bias = False
+    use_bias = True
 else:
     NormLayer_2d = nn.BatchNorm2d
     NormLayer_3d = nn.BatchNorm3d
-    use_bias = True
+    use_bias = False
 
 class encoder_2d(nn.Module):
     """Resnet-based generator that consists of Resnet blocks between a few downsampling/upsampling operations.

diff --git a/train/clean/train.py b/train/clean/train.py
@@ -23,17 +23,17 @@
 use_gpu = True
 use_gan = False
 use_L2 = True
-CONTINUE =  False
+CONTINUE =  True
 lambda_L1 = 100.0
 lambda_gan = 1
 
 SAVE_FRE = 10000
 start_iter = 0
 finesize = 128
 loadsize = int(finesize*1.1)
-batchsize = 1
+batchsize = 8
 perload_num = 16
-savename = 'MosaicNet_test'
+savename = 'MosaicNet_batch'
 dir_checkpoint = 'checkpoints/'+savename
 util.makedirs(dir_checkpoint)
 

diff --git a/util/util.py b/util/util.py
@@ -81,7 +81,7 @@ def get_bar(percent,num = 25):
     bar += ']'
     return bar+' '+str(round(percent,2))+'%'
 
-def copyfile(scr,dst):
+def copyfile(src,dst):
     try:
         shutil.copyfile(src, dst)
     except Exception as e: