Commit 7a436760 authored by yangzhong's avatar yangzhong
Browse files

更新README

parent 541e1080
数据集:ImageNet2012 validation
例:/public/opendas/DL_DATA/ImageNet-pytorch/val
使用模型,代码运行中自动下载resnet50开源模型:https://download.pytorch.org/models/resnet50-0676ba61.pth
使用resnet50开源模型,代码运行中自动下载:https://download.pytorch.org/models/resnet50-0676ba61.pth
单卡即可推理
测试方法:
export HIP_VISIBLE_DEVICES=0 # 指定卡号,不指定时默认使用第一张卡
python infer_resnet50.py
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