Spaces:
Runtime error
Runtime error
Gagan Bhatia
commited on
Commit
•
d818c99
1
Parent(s):
e284b45
Update model.py
Browse files- src/models/model.py +1 -0
src/models/model.py
CHANGED
@@ -13,6 +13,7 @@ from pytorch_lightning.callbacks.early_stopping import EarlyStopping
|
|
13 |
from pytorch_lightning import LightningDataModule
|
14 |
from pytorch_lightning import LightningModule
|
15 |
from datasets import load_metric
|
|
|
16 |
|
17 |
torch.cuda.empty_cache()
|
18 |
pl.seed_everything(42)
|
|
|
13 |
from pytorch_lightning import LightningDataModule
|
14 |
from pytorch_lightning import LightningModule
|
15 |
from datasets import load_metric
|
16 |
+
from tqdm.auto import tqdm
|
17 |
|
18 |
torch.cuda.empty_cache()
|
19 |
pl.seed_everything(42)
|