Example for Pytorch
·
DeepLearning
Import import numpy as np import pandas as pd import torch import torch.nn as nn import torch.nn.functional as F import torch.optim as optim 자주쓰는 기능들 optimizer.zero_grad() : 미분을 통해 얻은기울기 0 으로 초기화 cost.backward() : 비용함수를 미분하여 gradient계산 optimizer.step : 업데이트 단순 y = wx+b # 데이터 x_train = torch.FloatTensor([[1], [2], [3]]) y_train = torch.FloatTensor([[2], [4], [6]]) # 모델 초기화 W = torch.zeros(1, requ..
jjongguet
'pytorch' 태그의 글 목록