![](https://img-blog.csdnimg.cn/20201014180756754.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
codertanbo
这个作者很懒,什么都没留下…
展开
-
pytorch使用之微调网络模型
import torchvision.models as models 1.调整最后一层输出维度 model = models.ResNet(pretrained=True) fc_features = model.fc.in_features# 获取全连接层输入维度 model.fc = torch.nn.Linear(fc_features, num_class) 2.调整某一层参数 impo...原创 2019-04-28 19:29:29 · 3780 阅读 · 0 评论 -
pytorch使用之图像数据预处理
import torchvision from torchvision import datasets, transforms 一,创建一个预处理格式列表 transform_train_list = [ #transforms.RandomResizedCrop(size=128, scale=(0.75,1.0), ratio=(0.75,1.3333), interpol...原创 2019-04-29 15:06:48 · 2166 阅读 · 0 评论