Skip to content

Commit

Permalink
Add dataset_creator_name in config
Browse files Browse the repository at this point in the history
  • Loading branch information
Ming Du committed Jul 9, 2024
1 parent 45ff69e commit a8072f8
Showing 1 changed file with 8 additions and 0 deletions.
8 changes: 8 additions & 0 deletions generic_trainer/configs.py
Original file line number Diff line number Diff line change
Expand Up @@ -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.
Expand Down Expand Up @@ -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

Expand Down

0 comments on commit a8072f8

Please sign in to comment.