training_step
-
Pytorch Lightning 사용가이드 및 코드 예시(Scene Classification Dataset)Data Science/Pytorch 2022. 11. 17. 02:24
1. Pytorch Lightning이란? Tensorflow의 Keras와 같이 Pytorch를 위한 라이브러리이다. pytorch lightning의 전반적 동작 구조는 아래와 같다. 2. 라이브러리 준비 from glob import glob import torch import torch.utils.data as data from torchvision import transforms import PIL from torchvision.datasets import ImageFolder from torchvision.models import resnet18 from torch.optim import Adam from torch.nn import functional, CrossEntropyLoss from..