Commit 390999e4 authored by dcuai's avatar dcuai
Browse files

Update README.md

parent 6b97d166
...@@ -98,12 +98,15 @@ python train_v2.py configs/ms1mv2_r100 ...@@ -98,12 +98,15 @@ python train_v2.py configs/ms1mv2_r100
``` ```
torchrun --nproc_per_node=4 train_v2.py configs/ms1mv2_r100 torchrun --nproc_per_node=4 train_v2.py configs/ms1mv2_r100
``` ```
### 测试
## 精度
下载权重文件和测试数据集,测试模型精度: 下载权重文件和测试数据集,测试模型精度:
``` ```
python eval_ijbc.py --model-prefix model.pt --image-path IJBC_ALL --network r100 python eval_ijbc.py --model-prefix model.pt --image-path IJBC_ALL --network r100
``` ```
## result
### 精度
模型在MS1MV2数据集的测试指标: 模型在MS1MV2数据集的测试指标:
| 模型 | 数据类型 | AUC | | 模型 | 数据类型 | AUC |
| :------: | :------: | :------: | | :------: | :------: | :------: |
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment