Pytorch基于迁移学习的VGG卷积神经网络-手撕(可直接运行)-部分地方不懂的可以参考我...
test1=alexnet1(test1)print(test1.shape)epoch=2#迭代次数 learning=0.0001#学习率 optimizer=torch.optim.Adam(alexnet1.parameters(),lr=learning)#使用Adam优化器 loss=nn.CrossEntropyLoss()#损失计算方式,交叉熵 train_loss_all=[]#...