Remove the original coco preprocess. Fix a bug in train_rl_tb.py

miracle24 · May 5, 2017 · 3b7f1fe · 3b7f1fe
1 parent ab3b46c
commit 3b7f1fe
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 188 deletions.
diff --git a/coco/coco_preprocess.ipynb b/coco/coco_preprocess.ipynb
diff --git a/train_rl_tb.py b/train_rl_tb.py
@@ -65,6 +65,7 @@ def train(opt):
     # Assure in training mode
     model.train()
 
+    crit = utils.LanguageModelCriterion()
     rl_crit = utils.RewardCriterion()
 
     optimizer = optim.Adam(model.parameters(), lr=opt.learning_rate, weight_decay=opt.weight_decay)