search

Shreeyak / pytorch-lightning-segmentation-template

Semantic Segmentation on the LaPa dataset using Pytorch Lightning
Apache License 2.0
13 stars 5 forks source link

WIP: Wandb logging #16

Closed GainiK closed 3 years ago

GainiK commented 3 years ago

need to do the aggregation of results due to DDP

Shreeyak commented 3 years ago

Closing this in favor of #17 torch.distributed functions should not be required.

  1. 17 got losses to aggregate using self.log().

  2. For metrics, we'll try to use the PL Metrics API - subclass the Metrics class to implement custom metrics. See: https://pytorch-lightning.readthedocs.io/en/latest/metrics.html#implementing-a-metric