diff options
author | Jordan Gong <jordan.gong@protonmail.com> | 2021-01-23 00:16:58 +0800 |
---|---|---|
committer | Jordan Gong <jordan.gong@protonmail.com> | 2021-01-23 00:16:58 +0800 |
commit | f3321b6782aff20df2c029239cc0f8b9a59ffde7 (patch) | |
tree | e4b0c952d90e6ffcf255cdc4233496fb5612b886 /eval.py | |
parent | 7b5ff0a8dd2406af74776f31b1e2afabf9fadacd (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.py | 2 |
1 files changed, 1 insertions, 1 deletions
@@ -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}) |