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

fix some eval bugs

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