Gagan Bhatia commited on
Commit
171817a
1 Parent(s): 187bb52

Update model.py

Browse files
Files changed (1) hide show
  1. 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)