README.md 345 Bytes
Newer Older
yangzhong's avatar
yangzhong committed
1
2
3
4
5
6
7
数据集:ImageNet2012 validation
例:/public/opendas/DL_DATA/ImageNet-pytorch/val
使用模型,代码运行中自动下载resnet50开源模型:https://download.pytorch.org/models/resnet50-0676ba61.pth
单卡即可推理
测试方法:
export HIP_VISIBLE_DEVICES=0   # 指定卡号,不指定时默认使用第一张卡
python infer_resnet50.py