@@ -69,13 +69,13 @@ def main(args):
69
69
if __name__ == '__main__' :
70
70
parser = argparse .ArgumentParser ()
71
71
parser .add_argument ('--image' , type = str , required = True , help = 'input image for generating caption' )
72
- parser .add_argument ('--encoder_path' , type = str , default = 'models/encoder-2-1000.ckpt ' , help = 'path for trained encoder' )
73
- parser .add_argument ('--decoder_path' , type = str , default = 'models/decoder-2-1000.ckpt ' , help = 'path for trained decoder' )
72
+ parser .add_argument ('--encoder_path' , type = str , default = 'models/encoder-5-3000.pkl ' , help = 'path for trained encoder' )
73
+ parser .add_argument ('--decoder_path' , type = str , default = 'models/decoder-5-3000.pkl ' , help = 'path for trained decoder' )
74
74
parser .add_argument ('--vocab_path' , type = str , default = 'data/vocab.pkl' , help = 'path for vocabulary wrapper' )
75
75
76
76
# Model parameters (should be same as paramters in train.py)
77
77
parser .add_argument ('--embed_size' , type = int , default = 256 , help = 'dimension of word embedding vectors' )
78
78
parser .add_argument ('--hidden_size' , type = int , default = 512 , help = 'dimension of lstm hidden states' )
79
79
parser .add_argument ('--num_layers' , type = int , default = 1 , help = 'number of layers in lstm' )
80
80
args = parser .parse_args ()
81
- main (args )
81
+ main (args )
0 commit comments