Shreeyak / pytorch-lightning-segmentation-template

Semantic Segmentation on the LaPa dataset using Pytorch Lightning

Apache License 2.0

13 stars 5 forks source link

Save inference to disk #50

Closed Shreeyak closed 3 years ago

Shreeyak commented 3 years ago

Save results of inference to disk
Fixes bug where multiple log dirs created if gpus>1 in DDP.
Refactor - updated argument names, add clean methods to save on epoch and step.

Shreeyak commented 3 years ago

Fixes https://github.com/Shreeyak/pytorch-lightning-segmentation-template/issues/49 , https://github.com/Shreeyak/pytorch-lightning-segmentation-template/issues/36

Shreeyak commented 3 years ago

Fixes https://github.com/Shreeyak/pytorch-lightning-segmentation-template/issues/54

Shreeyak commented 3 years ago

54 will be a new PR for clarity.