PyTorch 使用心得
Posted fengyubo
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了PyTorch 使用心得相关的知识,希望对你有一定的参考价值。
PyTorch 使用心得
模板
import torch.nn as nn
import torch.optim as optim
class Model(nn.Module):
def __init__(self):
super(Model, self).__init__()
# self.hidden_layer = nn.Linear(in_dim, out_dim)
def forward(self, x):
# out = self.hidden_layer(x)
return out
model = Model()
optimizer = optim.Adam(model.parameters(), lr=lr)
criticism = nn.MSELoss()
for i in range(epoch):
optimizer.zero_grad()
y_pred = model(x)
loss = criticism(y_pred, y)
loss.backward()
optimizer.step()
# test
以上是关于PyTorch 使用心得的主要内容,如果未能解决你的问题,请参考以下文章