pytorch 关于DataParallel后模型的结构
关于DataParallel后模型的结构
import torch
import torch.nn as nn
import torch.optim as optim
# Define model
class TheModelClass(nn.Module):
def __init__(self):
super(TheModelClass, self).__init__()...
原创
2020-03-19 17:47:00 ·
1043 阅读 ·
0 评论