Spaces:
Runtime error
Runtime error
Gagan Bhatia
commited on
Commit
•
187bb52
1
Parent(s):
64226c2
Update model.py
Browse files- src/models/model.py +2 -0
src/models/model.py
CHANGED
@@ -13,3 +13,5 @@ import pytorch_lightning as pl
|
|
13 |
from pytorch_lightning.loggers import MLFlowLogger
|
14 |
from pytorch_lightning import Trainer
|
15 |
from pytorch_lightning.callbacks.early_stopping import EarlyStopping
|
|
|
|
|
|
13 |
from pytorch_lightning.loggers import MLFlowLogger
|
14 |
from pytorch_lightning import Trainer
|
15 |
from pytorch_lightning.callbacks.early_stopping import EarlyStopping
|
16 |
+
from pytorch_lightning import LightningDataModule
|
17 |
+
from pytorch_lightning import LightningModule
|