/pytorch_project_start

Files I use when I start a new Pytorch project

Primary LanguagePython

pytorch_project_start

Files I use when I start a new Pytorch project

features

runner class to run the training/testing loops

plotter generates learning curve plots

logger automatically logs the current training in a folder storing the model, metric on each iteration...