チラ裏備忘録

情報整理

2020-09-01から1ヶ月間の記事一覧

PyTorchで線形回帰(二次関数)

はじめに PyTorchで単純な線形回帰を行いたいと思います. nn.Linear()を使ってやるのもいいのですが,今回は,重みとバイアス用のTensorを自分で定義する方針で行いました. 色々読み込み import torch import torch.nn as nn import matplotlib.pyplot as …

PyTorchでMNIST

はじめに PyTorchを触ってみました. 試しに,単純なパーセプトロンのみでMNIST分類をやってみます.すべてGoogle Colabの環境で実行しました. Python: 3.6.9 PyTorch: 1.6.0+cu101 色々読み込み import torch, torch.nn as nn, torchvision from torchvisi…