Checkpoints - load weights - Githubissues

Shreeyak / pytorch-lightning-segmentation-template

Semantic Segmentation on the LaPa dataset using Pytorch Lightning

Apache License 2.0

13 stars 5 forks source link

Checkpoints - load weights #62

Closed Shreeyak closed 3 years ago

Shreeyak commented 3 years ago

Added new config group for loading weights
Added hyperparams to model
Some refactoring of trainer
Model accepts DictConfig and parses into dataclass in init.

Fixes https://github.com/Shreeyak/pytorch-lightning-segmentation-template/issues/18 Fixes https://github.com/Shreeyak/pytorch-lightning-segmentation-template/issues/40 Fixes https://github.com/Shreeyak/pytorch-lightning-segmentation-template/issues/51 Fixes https://github.com/Shreeyak/pytorch-lightning-segmentation-template/issues/52

Could not implement resuming training with wandb due to potential bug (#61)