2020年6月20日
Pytorch加载预训练模型前n层
摘要： import torch.nn as nn import torchvision.models as models class resnet(nn.Module): def __init__(self): super(resnet,self).__init__() self.model = mode 阅读全文
posted @ 2020-06-20 23:20 爱豆^2 阅读(1134) 评论(0) 推荐(0) 编辑