提交 6dd45118 编写于 作者: R root

fix some eval bugs

上级 d8def846
......@@ -9,4 +9,4 @@ TNT(Transformer-iN-Transformer)系列模型由华为诺亚于2021年提出
| Model | Params (M) | FLOPs (G) | Top-1 (%) | Top-5 (%) |
|:---------------------:|:----------:|:---------:|:---------:|:---------:|
| TNT_small | 23.8 | 5.2 | 81.12 | 95.56 |
\ No newline at end of file
| TNT_small | 23.8 | 5.2 | 81.21 | 95.63 |
......@@ -58,7 +58,9 @@ def build_dataloader(config, mode, device, seed=None):
'Train',
'Eval',
'Test',
], "Mode should be Train, Eval, Test"
'Gallery',
'Query'
], "Mode should be Train, Eval, Test, Gallery, Query"
# build dataset
config_dataset = config[mode]['dataset']
config_dataset = copy.deepcopy(config_dataset)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册