update evaluation

phoenix104104 · Jul 31, 2018 · f150be8 · f150be8
1 parent 193c94e
commit f150be8
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/batch_evaluate.py b/batch_evaluate.py
@@ -8,16 +8,13 @@
 
     ### model options
     parser.add_argument('-method',      type=str,       required=True,    help='full model name')
-    parser.add_argument('-epoch',       type=int,       default=-1,       help='model epoch')
     parser.add_argument('-gpu',         type=int,       default=0,        help='gpu device id')
     parser.add_argument('-metric',      type=str,       required=True,    choices=["LPIPS", "WarpError"])
     parser.add_argument('-redo',        action="store_true",              help='redo evaluation')
 
     opts = parser.parse_args()
     print(opts)
 
-    if opts.epoch != -1:
-        opts.method = os.path.join("output", opts.method, "epoch_%d" %opts.epoch)
 
     filename = "lists/test_tasks.txt"
     with open(filename) as f: