- 博客(2)
- 资源 (2)
- 收藏
- 关注
RSS订阅翻译 pytorch入门（2）
翻译自pytorch官方文档1.建立模型# Get cpu or gpu device for training.device = "cuda" if torch.cuda.is_available() else "cpu"print("Using {} device".format(device))# Define modelclass NeuralNetwork(nn.Module): def __init__(self): super(NeuralNetwo
2021-09-05 21:00:26
200
翻译 Pytorch入门
翻译自Pytorch官方英文文档1.定义并初始化神经网络import torchimport torch.nn as nnimport torch.optim as optimimport torch.nn.functional as Fclass Net(nn.Module): def __init__(self): super(Net, self).__init__() self.conv1 = nn.Conv2d(3, 6, 5)
2021-09-05 20:07:20
100
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人