摘要:
1、网络模型在pytorch里面的torchvision里面torchvision.models,是关于图像类的网络模型 2、简单以一个分类模型为例子: VGG(最常用的是VGG16和VGG19) pretrained: 如果是true的话,说明在ImageNet数据集上,模型的参数是都训练好的; 阅读全文
posted @ 2023-02-24 18:43
bokeAR
阅读(164)
评论(0)
推荐(0)
摘要:
1、optim的使用步骤: (1)构造优化器 (2)调用step方法 利用得到的梯度来对参数进行更新 loss_cross=CrossEntropyLoss() #损失cifar=CIFAR10_modle() #创建神经网络'''1、创建优化器'''optim=torch.optim.SGD(ci 阅读全文
posted @ 2023-02-24 13:36
bokeAR
阅读(89)
评论(0)
推荐(0)

浙公网安备 33010602011771号