没有合适的资源?快使用搜索试试~ 我知道了~
温馨提示
影像网 这将对ImageNet数据集上的流行模型体系结构(例如AlexNet,SqueezeNet,ResNet,DenseNet和VGG)进行训练(现在,我们支持alexnet,vgg,resnet,squeezenet,densenet)。 要求 PyTorch 0.4.0 cuda && cudnn 下载ImageNet数据集并将验证图像移动到带标签的子文件夹中 为此,您可以使用以下脚本: : 训练 要训练模型,请使用所需的模型架构和ImageNet数据集的路径运行main.py: python main.py [imagenet-folder with train and val folders] -a alexnet --lr 0.01 默认学习率计划从0.01开始,每30个时代衰减10倍。 用法 usage: main.py [-h] [-a ARCH] [--e
资源推荐
资源详情
资源评论
收起资源包目录
ImageNet-master.zip (14个子文件)
ImageNet-master
.gitignore 1KB
valprep.sh 2.12MB
README.md 6KB
LICENSE 1KB
.gitattributes 30B
data_loader.py 1KB
helper.py 1KB
models
resnet.py 7KB
alexnet.py 2KB
densenet.py 8KB
__init__.py 310B
vgg.py 7KB
squeezenet.py 4KB
main.py 9KB
共 14 条
- 1
资源评论
小旗旗
- 粉丝: 29
- 资源: 4557
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
最新资源
资源上传下载、课程学习等过程中有任何疑问或建议,欢迎提出宝贵意见哦~我们会及时处理!
点击此处反馈
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功