@@ -69,13 +69,13 @@ def main(args):
6969if  __name__  ==  '__main__' :
7070    parser  =  argparse .ArgumentParser ()
7171    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' )
7474    parser .add_argument ('--vocab_path' , type = str , default = 'data/vocab.pkl' , help = 'path for vocabulary wrapper' )
7575
7676    # Model parameters (should be same as paramters in train.py) 
7777    parser .add_argument ('--embed_size' , type = int  , default = 256 , help = 'dimension of word embedding vectors' )
7878    parser .add_argument ('--hidden_size' , type = int  , default = 512 , help = 'dimension of lstm hidden states' )
7979    parser .add_argument ('--num_layers' , type = int  , default = 1 , help = 'number of layers in lstm' )
8080    args  =  parser .parse_args ()
81-     main (args )
81+     main (args )
0 commit comments