- 博客(2)
- 资源 (2)
- 收藏
- 关注
翻译 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 model class NeuralNetwork(nn.Module): def __init__(self): super(NeuralNetwo
2021-09-05 21:00:26
153
翻译 Pytorch入门
翻译自Pytorch官方英文文档 1.定义并初始化神经网络 import torch import torch.nn as nn import torch.optim as optim import torch.nn.functional as F class Net(nn.Module): def __init__(self): super(Net, self).__init__() self.conv1 = nn.Conv2d(3, 6, 5)
2021-09-05 20:07:20
73
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人