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 使用心得的主要内容,如果未能解决你的问题,请参考以下文章

基于pytorch平台分类模型训练调式心得

pytorch 中实现CNN,对CNN的理解心得

pytorch 中实现CNN,对CNN的理解心得

深度学习及pytorch基础

暑假自学JAVA Web心得

Javalucene4.0学习心得