Try it here: https://github.com/DubiousCactus/matchbox
Follow the instructions on GitHub. Try the "builder" branch for the feature presented in this video demo.
About
Matchbox is: - a PyTorch project template - a set of utility functions for a broad range of deep learning projects - a bootstrapping boilerplate for datasets, models, training & testing logic with powerful configuration via hydra-zen - filled with safeguards for automatic and smart caching of data pre- and post- processing - a useful terminal UI for training/tesitng plots and loss curves for quick experimentation and monitoring (even through SSH sessions!) - a live coding environment with automatic code reloading and re-execution of localized logic