Pytorch
卓_尔_不_凡
这个作者很懒,什么都没留下…
展开
-
Pytorch学习---基于经典网络架构ResNet训练花卉图像分类模型
"""set_parameter_requires_grad 函数的作用是根据 feature_extracting 参数的值来决定是否冻结模型的参数。当用于特征提取时,它会阻止预训练模型的参数在训练过程中被更新,从而保留预训练模型的特征提取能力。当用于微调时,它不会修改参数的 requires_grad 属性,从而允许所有参数被更新。"""# 该函数会遍历模型的所有参数,并将它们的 requires_grad 属性设置为 False。原创 2024-09-20 22:43:46 · 679 阅读 · 0 评论 -
Pytorchz学习---基于卷积神经网络的MINIST数据集训练
# 一般情况下,如果模型有可学习的参数,最好用nn.Module,其他情况用nn.function相对简单一些import torch.nn.functional as Fimport torchloss_func = F.cross_entropydef model(xb): return xb.mm(weights)+bias# didi卷积过程# bs = 64# xb = x_trains[0:bs]# yb = y_trains[0:bs]# weights =原创 2024-09-18 17:10:59 · 600 阅读 · 0 评论