swd543 / torch-boilerplate

Very minimal pytorch boilerplate with wandb logging and multi gpu support

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

torch-boilerplate

Pytorch boilerplate with wandb logging and multi gpu support. Very minimal template with running metrics (training and validation accuracies) and support for lr_scheduler.

Todo

  • Model checkpointing and loading
  • Early stopping

About

Very minimal pytorch boilerplate with wandb logging and multi gpu support

License:MIT License


Languages

Language:Python 100.0%