Lightning-AI / deep-learning-project-template

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

Encourage Modularity of Lightning Modules #19

Open teddykoker opened 3 years ago

teddykoker commented 3 years ago

Would be great if people's lightning modules became easily accessible via pip install. This functionality may work as is right now, but it would be great to showcase and encourage this pattern:

pip install git+https://github.com/teddykoker/teddys-model/

Then in python

from pl_bolts.datamodules import ImageNet
from teddys-model import TeddysModel

pl.Trainer().fit(TeddysModel(), ImageNet())

This encourages reusability of modules and datamodules which would be great for research reproducibility as well as just plug-n-play with different models.

teddykoker commented 3 years ago

cc @nateraw