summaryrefslogtreecommitdiff
path: root/eval.py
diff options
context:
space:
mode:
authorJordan Gong <jordan.gong@protonmail.com>2021-01-23 00:16:58 +0800
committerJordan Gong <jordan.gong@protonmail.com>2021-01-23 00:16:58 +0800
commitf3321b6782aff20df2c029239cc0f8b9a59ffde7 (patch)
treee4b0c952d90e6ffcf255cdc4233496fb5612b886 /eval.py
parent7b5ff0a8dd2406af74776f31b1e2afabf9fadacd (diff)
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
Diffstat (limited to 'eval.py')
-rw-r--r--eval.py2
1 files changed, 1 insertions, 1 deletions
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})