首页 > 解决方案 > 在pytroch中对预训练模型的参数进行改组?

问题描述

我在 Pytorch 中导入了一个预训练的 resnet50,我想重新调整权重,怎么做?我已经这样做了-

    model_conv = torchvision.models.resnet50(pretrained=True)
                                                                                                    
    for param in model_conv.parameters():
       torch.utils.data.DataLoader(dataset=param, shuffle=True)

标签: pythonmachine-learningdeep-learningpytorch

解决方案


推荐阅读