Skip to content

Commit 82b0bb0

Browse files
authored
Update train.lua
1 parent 45582da commit 82b0bb0

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

train.lua

+1-1
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@ cmd:option('-feature_type', 'VGG', 'VGG or Residual')
4040

4141
cmd:option('-hidden_size',512,'the hidden layer size of the model.')
4242
cmd:option('-rnn_size',512,'size of the rnn in number of hidden nodes in each layer')
43-
cmd:option('-batch_size',20,'what is theutils batch size in number of images per batch? (there will be x seq_per_img sentences)')
43+
cmd:option('-batch_size',300,'what is theutils batch size in number of images per batch? (there will be x seq_per_img sentences)')
4444
cmd:option('-output_size', 1000, 'number of output answers')
4545
cmd:option('-rnn_layers',2,'number of the rnn layer')
4646

0 commit comments

Comments
 (0)