Spaces:
Runtime error
Runtime error
Gagan Bhatia commited on
Commit ·
171817a
1
Parent(s): 187bb52
Update model.py
Browse files- src/models/model.py +3 -0
src/models/model.py
CHANGED
|
@@ -15,3 +15,6 @@ from pytorch_lightning import Trainer
|
|
| 15 |
from pytorch_lightning.callbacks.early_stopping import EarlyStopping
|
| 16 |
from pytorch_lightning import LightningDataModule
|
| 17 |
from pytorch_lightning import LightningModule
|
|
|
|
|
|
|
|
|
|
|
|
| 15 |
from pytorch_lightning.callbacks.early_stopping import EarlyStopping
|
| 16 |
from pytorch_lightning import LightningDataModule
|
| 17 |
from pytorch_lightning import LightningModule
|
| 18 |
+
|
| 19 |
+
torch.cuda.empty_cache()
|
| 20 |
+
pl.seed_everything(42)
|