From c67bc95d771b55ee78a9ec517e393a46f55c9558 Mon Sep 17 00:00:00 2001
From: FangmingChou <37794814+FangmingChou@users.noreply.github.com>
Date: Sun, 3 Nov 2019 16:57:28 +0800
Subject: [PATCH] Update test.py

the argment "checkpoint_dir" should be "model_path"
---
 test.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/test.py b/test.py
index cd82c6c..e426ed3 100644
--- a/test.py
+++ b/test.py
@@ -75,7 +75,7 @@ def main(args):
             ac_i2t_top10_best = ac_top10_i2t
             ac_t2i_top1_best = ac_top1_t2i
             ac_t2i_top10_best = ac_top10_t2i
-            dst_best = os.path.join(args.checkpoint_dir, 'model_best', str(epoch)) + '.pth.tar'
+            dst_best = os.path.join(args.model_path, 'model_best', str(epoch)) + '.pth.tar'
             shutil.copyfile(model_file, dst_best)
          
         logging.info('epoch:{}'.format(epoch))