IBOK

기본편 - 자동 저장 Saving and loading checkpoints (basic) — PyTorch Lightning 1.9.0 documentation Shortcuts pytorch-lightning.readthedocs.io PyTorch Lightning의 Trainer을 이용해 학습을 진행하면, 자동으로 가장 마지막 training epoch의 checkpoint를 저장해준다. trainer = Trainer() 만약 checkpoint가 저장되는 위치를 바꾸고 싶다면 다음과 같이 지정해줄 수 있다. trainer = Trainer(default_root_dir='path/to/') 혹은 별도로 checkpoint 저장을 하지 않으려면 다음과 같이 지정하면 된다. trainer = Tr..

Logging — PyTorch Lightning 1.8.6 documentation Shortcuts pytorch-lightning.readthedocs.io 로그를 기록하는 방법은 Lightening Module에서 self.log()나 self.log_dict()를 이용하면 된다. def training_step(self, batch, batch_idx): self.log_dict({'acc': acc, 'recall': recall}) self.log('acc', acc) logging (log(), log_dict() 모두 동일하게 적용) 의 중요한 인자는 on_step과 on_epoch이다. on_step: 현재 step에 logging on_epoch: 로그를 축적하여 epoch 마지막에 ..