速習 pytorch-lightning: 今すぐ機械学習の実験をしたいそこのキミへ - Qiita

速習 pytorch-lightning: 今すぐ機械学習の実験をしたいそこのキミへ - Qiita

概要 pytorch lightningは、PyTorchでの開発スピードを爆速にしてくれるライブラリです。今回はこのライ...概要を表示 概要 pytorch lightningは、PyTorchでの開発スピードを爆速にしてくれるライブラリです。今回はこのライブラリを使ってサクッとCNNを実装していきます。 コード import os import torch import torch.nn as nn import torch.nn.functional as F from torch.utils.data import DataLoader from torchvision.datasets import MNIST from torchvision import transforms import pytorch_lightning as ptl class CNN(ptl.LightningModule): # モデルの定義(PyTorchと一緒) def __init__(self): super(CNN, sel

• あとで読む