From dbf79775be5bfb73fdc1170454a89dd09d59375b Mon Sep 17 00:00:00 2001 From: chaoyan Date: Sat, 28 Jul 2018 15:45:07 -0400 Subject: [PATCH] fix bugs when testing --- evaluate_model.ipynb | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/evaluate_model.ipynb b/evaluate_model.ipynb index 499f087..4e6166f 100644 --- a/evaluate_model.ipynb +++ b/evaluate_model.ipynb @@ -60,7 +60,7 @@ "outputs": [], "source": [ "model = CaptionGenerator(word_to_idx, dim_feature=[196, 512], dim_embed=512,\n", - " dim_hidden=1500, n_time_step=16, prev2out=True, \n", + " dim_hidden=1024, n_time_step=16, prev2out=True, \n", " ctx2out=True, alpha_c=1.0, selector=True, dropout=True)" ] }, @@ -74,7 +74,7 @@ "source": [ "solver = CaptioningSolver(model, data, data, n_epochs=15, batch_size=128, update_rule='adam',\n", " learning_rate=0.0025, print_every=2000, save_every=1, image_path='./image/val2014_resized',\n", - " pretrained_model=None, model_path='./model/lstm', test_model='./model/lstm3/model-18',\n", + " pretrained_model=None, model_path='./model/lstm', test_model='./model/lstm/model-18',\n", " print_bleu=False, log_path='./log/')" ] },