torch.nn.DataParallel使用细节
仅使用nn.DataParallel，gpu0和gpu1、gpu0和gpu2、gpu0和gpu3等包含gpu0的组合都是可以的，其余组合不行，报错RuntimeError: module must h ...
花费 45 ms
Pytorch固定某些层的操作
1. model = models.resnet18(pretrained=False,num_classes=CIFAR10_num_classes) def my_forward(model, x ...