From f3321b6782aff20df2c029239cc0f8b9a59ffde7 Mon Sep 17 00:00:00 2001
From: Jordan Gong <jordan.gong@protonmail.com>
Date: Sat, 23 Jan 2021 00:16:58 +0800
Subject: Evaluation bug fixes and code review

1. Return full cached clip in evaluation
2. Add multi-iter checkpoints support in evaluation
3. Remove duplicated code while transforming
---
 eval.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'eval.py')

diff --git a/eval.py b/eval.py
index 7b68220..ec48066 100644
--- a/eval.py
+++ b/eval.py
@@ -14,7 +14,7 @@ dataset_selectors = {
     'cl': {'conditions': ClipConditions({r'nm-0\d', r'cl-0\d'})},
 }
 
-accuracy = model.predict_all(config['model']['total_iter'], config['dataset'],
+accuracy = model.predict_all(config['model']['total_iters'], config['dataset'],
                              dataset_selectors, config['dataloader'])
 rank = 5
 np.set_printoptions(formatter={'float': '{:5.2f}'.format})
-- 
cgit v1.2.3