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