はてなブックマーク

概要 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