Pytorch boilerplate with wandb logging and multi gpu support. Very minimal template with running metrics (training and validation accuracies) and support for lr_scheduler.
- Model checkpointing and loading
- Early stopping
swd543/torch-boilerplate
Very minimal pytorch boilerplate with wandb logging and multi gpu support
PythonMIT