首页 > 其他 > 详细

PyTorch 使用心得

时间:2018-07-04 23:23:49      阅读:250      评论:0      收藏:0      [点我收藏+]

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 使用心得

原文:https://www.cnblogs.com/fengyubo/p/9265844.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!