位置: pytorch自定义网络层 - 标签
pytorch对网络层的增,删, 改, 修改预训练模型结构(pytorch自定义网络层)
简述:pytorch对网络层的增,删,改,修改预训练模型结构#下载模型参数model.load_state_dict(torch.load('model.pth'))#再加载网络的参数torch.load('model.pth')是获
发布时间:2024/01/17
鄂ICP备2023003026号