This repository is an extension meant to simplify Population Based Training using PyTorch Lightning.
Currently we are implementing this as a standalone project, but most all of the code is based directly on PyTorch Lightning and we hope to integrate this into that package as we develop better testing and integration.
This project is still very much in a development phase, but if you'd like to try it out we've included some setup instructions below, and some starter examples here.
Currently this implementation supports the truncation method and performs perturbations on user-defined hyperparameters.
# core package
pip install -e .
# examples
pip install -e .pytorch_lightning_pbt_examples/