diff --git a/opts.lua b/opts.lua index 5d86d6e..5669117 100644 --- a/opts.lua +++ b/opts.lua @@ -33,13 +33,13 @@ function opts.parse(arg) cmd:option('--n_threads', 4, 'Number of threads for multi-thread loading') cmd:option('--checkpoint', 10, 'Trained model is saved at every this epochs.') cmd:option('--manual_seed', 0, 'Manually set random seed') - cmd:option('--crop', 'r', 'Spatial cropping method. random is uniform. corner is selection from 4 corners and 1 cetner. (r | c)') + cmd:option('--crop', 'c', 'Spatial cropping method. random is uniform. corner is selection from 4 corners and 1 cetner. (r | c)') cmd:option('--no_hflip', false, 'If true holizontal flipping is not performed.') cmd:option('--batch_norm', false, 'If true batch normalization is added to C3D') cmd:option('--global_pooling', false, 'If true global average pooling is added to C3D') cmd:option('--model', 'c3d', 'Network model (c3d | resnet)') - cmd:option('--resnet_depth', 18, 'Depth of resnet') - cmd:option('--shortcut_type', 'A', 'Shortcut type of resnet') + cmd:option('--resnet_depth', 18, 'Depth of resnet (10 | 18 | 34 | 50 | 101 | 152)') + cmd:option('--shortcut_type', 'A', 'Shortcut type of resnet (A | B | C)') cmd:option('--weight_init', false, 'Weight initialization for resnet') cmd:option('--test_subset', 'val', 'Subset to use for test_video (val | test)') cmd:option('--freeze_params', 0, '0 means all layers are trained. 1 means last layer is trained.')