chopsize

yulunzhang · Apr 2, 2018 · b72cd88 · b72cd88
1 parent 5b733b7
commit b72cd88
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/RDN_TrainCode/code/opts.lua b/RDN_TrainCode/code/opts.lua
@@ -63,7 +63,7 @@ function M.parse(arg)
     cmd:option('-skipBatch',        3,                  'Skip mini-batch if the error is bigger than skipBath * (previous error), 2')
     cmd:option('-reset',            'false',            'Reset training')
     cmd:option('-chopShave',        10,                 'Shave width for chopForward')
-    cmd:option('-chopSize',         2e4,               'Minimum chop size for chopForward, 16e4')
+    cmd:option('-chopSize',         1e4,               'Minimum chop size for chopForward, 16e4')
     cmd:option('-naiveForward',     'false',            'Simple forward when test, false')
     -- Optimization
     cmd:option('-optimMethod',      'ADAM',             'Optimization method')

diff --git a/RDN_TrainCode/code/opts.lua~ b/RDN_TrainCode/code/opts.lua~
@@ -16,6 +16,7 @@ function M.parse(arg)
     cmd:option('-nGPU',             1,                  'Number of GPUs to use by default')
     cmd:option('-gpuid',            1,                  'GPU id to use')
     cmd:option('-nThreads',         3,                  'Number of data loading threads')
+    cmd:option('-save',             now,                'Subdirectory to save/log experiments in')
     cmd:option('-datadir',          '/home/yulun/data', 'Dataset location')
     cmd:option('-dataset',          'div2k',            'Dataset for training: div2k | flickr2k | SR291')
     cmd:option('-augRF',            'true',             'Use x8 augmented DVI2K LR train set, RF')