Spaces:
Runtime error
Runtime error
Gagan Bhatia
commited on
Commit
•
336121a
1
Parent(s):
ea2c6ba
Update model.py
Browse files- src/models/model.py +3 -0
src/models/model.py
CHANGED
|
@@ -24,3 +24,6 @@ class DataModule(Dataset):
|
|
| 24 |
"""
|
| 25 |
Data Module for pytorch
|
| 26 |
"""
|
|
|
|
|
|
|
|
|
|
|
|
| 24 |
"""
|
| 25 |
Data Module for pytorch
|
| 26 |
"""
|
| 27 |
+
|
| 28 |
+
def __init__(
|
| 29 |
+
self,
|