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 |
|