From a8072f8b1b2bc1ff929eadd1769a43935f06a658 Mon Sep 17 00:00:00 2001 From: Ming Du Date: Tue, 9 Jul 2024 11:40:57 -0500 Subject: [PATCH] Add dataset_creator_name in config --- generic_trainer/configs.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/generic_trainer/configs.py b/generic_trainer/configs.py index 3bc2908..6ee666c 100644 --- a/generic_trainer/configs.py +++ b/generic_trainer/configs.py @@ -94,6 +94,7 @@ def load_from_json(self, filename, namespace=None): def string_to_object(self, key, value): """ Create an object based on the string value of a config key. + :param key: str. :param value: str. :return: object. @@ -229,6 +230,13 @@ class Config(OptionContainer): - 'cls': a classification prediction. - 'regr': a regression prediction. """ + + dataset_creator_name: Optional[str] = None + """ + Name of the data creator function. This parameter is not used by the trainer itself, but the training + script can use this parameter to locate a custom data creator function that creates the training, + va + """ debug: bool = False