cleaning up demo file
This commit is contained in:
parent
5c56295421
commit
64bdd1c46d
|
@ -20,11 +20,6 @@ np.random.seed(SEED)
|
|||
from lightning_module_template import LightningTemplateModel
|
||||
# ---------------------
|
||||
|
||||
AVAILABLE_MODELS = {
|
||||
'model_template': LightningTemplateModel
|
||||
}
|
||||
|
||||
|
||||
"""
|
||||
Allows training by using command line arguments
|
||||
Run by:
|
||||
|
|
Loading…
Reference in New Issue