摘要: 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)