Spaces:
Runtime error
Runtime error
Gagan Bhatia
commited on
Commit
•
e284b45
1
Parent(s):
9f3cfcd
Update model.py
Browse files- src/models/model.py +1 -0
src/models/model.py
CHANGED
@@ -7,6 +7,7 @@ from transformers import (
|
|
7 |
)
|
8 |
from torch.utils.data import Dataset, DataLoader
|
9 |
import pytorch_lightning as pl
|
|
|
10 |
from pytorch_lightning import Trainer
|
11 |
from pytorch_lightning.callbacks.early_stopping import EarlyStopping
|
12 |
from pytorch_lightning import LightningDataModule
|
|
|
7 |
)
|
8 |
from torch.utils.data import Dataset, DataLoader
|
9 |
import pytorch_lightning as pl
|
10 |
+
from pytorch_lightning.loggers import MLFlowLogger, WandbLogger
|
11 |
from pytorch_lightning import Trainer
|
12 |
from pytorch_lightning.callbacks.early_stopping import EarlyStopping
|
13 |
from pytorch_lightning import LightningDataModule
|