Skip to content

Commit 5ea3ee3

Browse files
committed
Register datasets from utils
1 parent 850f762 commit 5ea3ee3

1 file changed

Lines changed: 6 additions & 0 deletions

File tree

src/utils.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,12 @@
55
import pytorch_lightning as L
66
from torch import nn
77

8+
from data import DATASETS
9+
10+
11+
for data_name, data in DATASETS.items():
12+
gin.external_configurable(data, data_name)
13+
814

915
def gin_config_to_readable_dictionary(gin_config: dict):
1016
"""

0 commit comments

Comments
 (0)