Change as train, model files

Author: capimx

tutorials/03-advanced/image_captioning/sample.py

@@ -35,7 +35,7 @@ def main(args):
 
     # Build models
     encoder = EncoderCNN(args.embed_size).eval()  # eval mode (batchnorm uses moving mean/variance)
-    decoder = DecoderRNN(args.embed_size, args.hidden_size, len(vocab), args.num_layers)
+    decoder = DecoderRNN(args.embed_size, args.hidden_size, vocab, args.num_layers)
     encoder = encoder.to(device)
     decoder = decoder.to(device)