Gagan Bhatia commited on
Commit
ea2c6ba
·
1 Parent(s): 74e821f

Update model.py

Browse files
Files changed (1) hide show
  1. src/models/model.py +3 -0
src/models/model.py CHANGED
@@ -21,3 +21,6 @@ pl.seed_everything(42)
21
 
22
 
23
  class DataModule(Dataset):
 
 
 
 
21
 
22
 
23
  class DataModule(Dataset):
24
+ """
25
+ Data Module for pytorch
26
+ """