Spaces:
Runtime error
Runtime error
Gagan Bhatia
commited on
Commit
•
7437ef5
1
Parent(s):
d818c99
Update model.py
Browse files- src/models/model.py +3 -0
src/models/model.py
CHANGED
@@ -15,6 +15,9 @@ 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)
|
20 |
|
|
|
15 |
from datasets import load_metric
|
16 |
from tqdm.auto import tqdm
|
17 |
|
18 |
+
# from dagshub.pytorch_lightning import DAGsHubLogger
|
19 |
+
|
20 |
+
|
21 |
torch.cuda.empty_cache()
|
22 |
pl.seed_everything(42)
|
23 |
|