Skip to content

Commit

Permalink
Sets default max epochs to 100; explicitly bool check args.download i…
Browse files Browse the repository at this point in the history
…n prepro
  • Loading branch information
abhshkdz committed May 27, 2018
1 parent 7e3f418 commit 4dd5571
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion data/prepro.py
Expand Up @@ -187,7 +187,7 @@ def get_image_ids(data, dtype):
if __name__ == "__main__":
args = parser.parse_args()

if args.download == 1:
if args.download == True:
os.system('wget https://computing.ece.vt.edu/~abhshkdz/data/visdial/visdial_0.9_train.zip')
os.system('wget https://computing.ece.vt.edu/~abhshkdz/data/visdial/visdial_0.9_val.zip')
# if args.train_split == 'trainval':
Expand Down
2 changes: 1 addition & 1 deletion opts.lua
Expand Up @@ -32,7 +32,7 @@ cmd:option('-batchSize', 40, 'Batch size (number of threads) (Adjust base on GPU
cmd:option('-learningRate', 1e-3, 'Learning rate')
cmd:option('-weightInit', 'xavier', 'Weight initialization strategy: xavier|heuristic|kaiming')
cmd:option('-dropout', 0.5, 'Dropout')
cmd:option('-numEpochs', 400, 'Epochs')
cmd:option('-numEpochs', 100, 'Epochs')
cmd:option('-LRateDecay', 10, 'After lr_decay epochs lr reduces to 0.1*lr')
cmd:option('-lrDecayRate', 0.9997592083, 'Decay for learning rate')
cmd:option('-minLRate', 5e-5, 'Minimum learning rate')
Expand Down

0 comments on commit 4dd5571

Please sign in to comment.