ML & DL
2021. 4. 8.
[DL] Pytorch Lightning ⚡
번개 파이토치에 대해 알아보자~ 파이토치 라이트닝은 기존 파이토치에서 학습/추론 및 데이터 로드 부분을 모듈화하여 반복되는 코드를 구현하지 않아도 되는 패키지다. 기존 파이토치는 모델 클래스, 데이터셋 클래스, 학습 과정을 따로 구현했으나 라이트닝은 한 클래스 안에 전부 구현한다. Gru를 이용한 오토인코더 모델을 예시로 구현해봤다. import pytorch_lightning as pl import torch import torch.nn as nn import torch.nn.functional as F class GruAutoencoder(pl.LightningModule): def __init__(self, config, chord_token, batch_size): super().__init__(..
