README.md

数据集: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

# 8卡推理
self.batch_size = 128 * 8     # 修改卡数

export HIP_VISIBLE_DEVICES=0,1,2,3,4,5,6,7

python infer_resnet50_8.py