A training pipeline for neural nets in PyTorch
A pipeline for training neural nets using PyTorch-lightning and hydra.
This is my training pipeline for training neural nets in PyTorch. It is based on PyTorch-lightning as a wrapper over PyTorch code and hydra for managing configuration files.