Lightning-AI / deep-learning-project-template

Pytorch Lightning code guideline for conferences
Apache License 2.0
1.23k stars 269 forks source link

Generic extension of this template with Hydra configs #23

Closed ashleve closed 3 years ago

ashleve commented 3 years ago

Recently I've been trying to develop a very general PyTorch Lightning + Hydra template.

The idea is to extend deep-learning-project-template, by just adding some Hydra config structure and simple example, which initializes modules from config parts in the most generic and obvious way.

I've been wondering if it would be a good idea to develop an official template of this sort and add it to Lightning ecosystem as an alternative for deep-learning-project-template? Currently it seems like Hydra is still a little rough around the edges and it's not obvious how config should be structured for Lightning, but still such a template could be a useful addition in the future.

Here is a link to my version: https://github.com/hobogalaxy/lightning-hydra-template