Generic Main

[nateraw]

🚀 Feature

Once DMs have been decoupled from models (#207) + we have map of aliases-to-classes (#201), we can put together a generic_main function that'll work for 90% of the models in this repo.

Motivation

• Makes running experiments really easy
• Allows us to parameterize tests super elegantly (1 test function to rule them all that is parameterized w/ pytest).

Pitch

something along the lines of this from the command line...

pl_bolts run --model vae --datamodule mnist --...any dm or model flag...

or this from python

from pl_bolts import bolts_main

bolts_main(model='vae', datamodule='mnist', **kwargs)

Alternatives

Additional context

Its significantly more complex to do this generically within lightning itself, but since we have control over interface of dms in this repo, we can cut a few corners to make it happen here for now while we work out the kinks for a generic main in lightning.