Dice score를 대상으로 동작하는 코드 BraTS2020 데이터 셋을 가지고 동작을 한다. 코드 # lighting.py import torch import torch.nn as nn import torch.optim import pytorch_lightning as pl from monai.losses.dice import DiceLoss, DiceFocalLoss from monai.metrics import DiceMetric, ConfusionMatrixMetric, HausdorffDistanceMetric from monai.utils.enums import MetricReduction from monai.data.utils import decollate_batch from torchme..

본 게시글에 작성되는 내용은 PyTorch Lightning 공식 사이트에 있는 Tutorial과 API Docs를 참고하여 작성하였습니다. 그 외에 블로그에 작성된 몇 개의 예제들을 더 참고했는데, PyTorch-Lightning이 추구하는 '효과적인 추상화'를 위해서는 이러한 방향으로 사용하면 더 좋을 수 있겠다 싶어서 글로 공부했던 내용을 글로 정리했습니다. PyTorch Lightning 소개 PyTorch Llightning은 PyTroch에 대한 High-level (abstract) 인터페이스를 제공하는 오픈소스 라이브러리 입니다. PyTorch 하나만으로도 숙달되면 부족함은 전혀 없지만, 모델이 커질 수록 그리고 다양하고 고도화된 기술들을 적용하면서 실험을 할 경우 코드가 너무 복잡해집니다..