pytorch中nn.DataParallel多次使用

本文主要是介绍pytorch中nn.DataParallel多次使用，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

pytorch中nn.DataParallel多次使用

import torch
import torch.nn as nn
import torch.optim as optim
from torch.utils.data import DataLoader# 定义模型
class MyModel(nn.Module):def __init__(self):super(MyModel, self).__init__()self.fc = nn.Linear(10, 1)def forward(self, x):return self.fc(x)# 创建模型实例
model = MyModel()# 将模型放到多张GPU上
model = nn.DataParallel(model)
model_witoout_dp = model.module

这篇关于pytorch中nn.DataParallel多次使用的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---