deep-learning-project-template icon indicating copy to clipboard operation
deep-learning-project-template copied to clipboard

Encourage Modularity of Lightning Modules

Open teddykoker opened this issue 5 years ago • 1 comments

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 avatar Oct 27 '20 20:10 teddykoker

cc @nateraw

teddykoker avatar Oct 27 '20 20:10 teddykoker