摘要:
一、官网内容 模型 数据集 二、案例 分类模型:Vgg16 1.参数 pretrained:True->已经训练过的模型,并能在数据集上取得一个比较好的效果。 progress:True->显示下载进度条 2.代码 2.1 下载数据集ImageNet 下载需要的包 pip install scipy 阅读全文
posted @ 2022-05-24 22:05
Trouvaille_fighting
阅读(73)
评论(0)
推荐(0)
摘要:
一、官方文档 torch.optimizer 优化器的定义 之后调用优化器的step方法 作用:就是利用得到的梯度对参数进行更新 相关优化器的算法 共同点:模型参数,学习速率,优化器自身的参数 二、案例 1. 代码 学习速率的设置:太大-模型不稳定,太小-速率太慢,一般由大到小设置 多个epoch进 阅读全文
posted @ 2022-05-24 11:21
Trouvaille_fighting
阅读(57)
评论(0)
推荐(0)

浙公网安备 33010602011771号