algorithm/pytorch-tutorial
1
0
Fork 0
mirror of https://github.com/yunjey/pytorch-tutorial.git synced 2026-03-13 09:11:37 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #30 from keishinkickback/master

Browse Source 
Bug fixes in /tutorials/09 - Image Captioning/train.py
This commit is contained in:
yunjey
2017-05-06 22:55:47 +09:00
committed by GitHub
parent 25ce8300ce b2c7bd7bfc
commit 12afd0b0f2
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tutorials/09 - Image Captioning/train.py
View File

@@ -3,6 +3,7 @@ import torch
import torch.nn as nn
import numpy as np
import os
import pickle
from data_loader import get_loader
from build_vocab import Vocabulary
from model import EncoderCNN, DecoderRNN
@@ -24,7 +25,7 @@ def main(args):
transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))])
# Load vocabulary wrapper.
with open(vocab_path, 'rb') as f:
with open(args.vocab_path, 'rb') as f:
vocab = pickle.load(f)
# Build data loader
@@ -115,4 +116,3 @@ if __name__ == '__main__':
parser.add_argument('--learning_rate', type=float, default=0.001)
args = parser.parse_args()
print(args)
main(args)