(2024-03-14) Pytorch Lightning
Pytorch Lightning을 이용한 CNN Classifier class CNNClassifier(pl.LightningModule): def __init__(self): super(CNNClassifier, self).__init__() def forward(self, x): def configure_optimizers(self): def training_step(self, batch, batch_idx): def validation_step(self, batch, batch_idx): def test_step(self, batch, batch_idx): def predict_step(self, batch, batch_idx): model = CNNClassifier(num_classes=10, ..
2024. 3. 14.