2020年8月14日
Pytorch固定某些层的操作
摘要： 1. model = models.resnet18(pretrained=False,num_classes=CIFAR10_num_classes) def my_forward(model, x): mo = nn.Sequential(*list(model.children())[:-1] 阅读全文
posted @ 2020-08-14 09:11 zonechen 阅读(712) 评论(0) 推荐(0) 编辑