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