2017年12月17日
pytorch预训练模型
摘要： 1.加载预训练模型： 只加载模型，不加载预训练参数：resnet18 = models.resnet18(pretrained=False) print resnet18 打印模型结构 resnet18.load_state_dict(torch.load('resnet18-5c106cde.pt 阅读全文
posted @ 2017-12-17 23:13 zhanghouyu 阅读(11872) 评论(0) 推荐(0) 编辑