举一个很简单的例子来熟悉使用pytorch构建和训练神经网络的整个过程。
import torch
import torch.nn as nn
N, D_in, H, D_out = 64, 1000, 100, 10
#随机创建一些训练数据
x = torch.randn(N, D_in)
y = torch.randn(N, D_out)
#定义model
class TowLayerNet(nn.Module
举一个很简单的例子来熟悉使用pytorch构建和训练神经网络的整个过程。
import torch
import torch.nn as nn
N, D_in, H, D_out = 64, 1000, 100, 10
#随机创建一些训练数据
x = torch.randn(N, D_in)
y = torch.randn(N, D_out)
#定义model
class TowLayerNet(nn.Module