if use_gpu:model = torch.nn.DataParallel(model, device_ids=[0, 1])#####加上这句model = model.cuda()model = load_network(model)########再加载参数
已解决
if use_gpu:model = torch.nn.DataParallel(model, device_ids=[0, 1])#####加上这句model = model.cuda()model = load_network(model)########再加载参数
已解决