1. 首页
  2. 数据库
  3. 其它
  4. Pytorch多GPU的计算和Sync BatchNorm

Pytorch多GPU的计算和Sync BatchNorm

上传者: 2021-01-16 14:00:28上传 PDF文件 83.55KB 热度 15次
nn.DataParallel pytorch中使用GPU非常方便和简单: import torch import torch.nn as nn input_size = 5 output_size = 2 class Model(nn.Module): def __init__(self, input_size, output_size): super(Model, self).__init__() self.fc = nn.Linear(input_size, output_size) def forward(self, input):
下载地址
用户评论