Spaces:
Runtime error
Runtime error
Gagan Bhatia commited on
Commit ·
74e821f
1
Parent(s): 171817a
Update model.py
Browse files- src/models/model.py +3 -0
src/models/model.py
CHANGED
|
@@ -18,3 +18,6 @@ from pytorch_lightning import LightningModule
|
|
| 18 |
|
| 19 |
torch.cuda.empty_cache()
|
| 20 |
pl.seed_everything(42)
|
|
|
|
|
|
|
|
|
|
|
|
| 18 |
|
| 19 |
torch.cuda.empty_cache()
|
| 20 |
pl.seed_everything(42)
|
| 21 |
+
|
| 22 |
+
|
| 23 |
+
class DataModule(Dataset):
|