ertk.pytorch.train.PyTorchDataConfig
- class ertk.pytorch.train.PyTorchDataConfig(train_data_processing: dict[str, ertk.pytorch.train.PyTorchDataAugConfig] = <factory>, valid_data_processing: dict[str, ertk.pytorch.train.PyTorchDataAugConfig] = <factory>, test_data_processing: dict[str, ertk.pytorch.train.PyTorchDataAugConfig] = <factory>)
Bases:
ERTKConfig- __init__(train_data_processing: dict[str, ertk.pytorch.train.PyTorchDataAugConfig] = <factory>, valid_data_processing: dict[str, ertk.pytorch.train.PyTorchDataAugConfig] = <factory>, test_data_processing: dict[str, ertk.pytorch.train.PyTorchDataAugConfig] = <factory>) None
Methods
__init__([train_data_processing, ...])Inherited Methods
default()Create default config.
from_config(config)Create config object from any compatible config.
from_file(path)Create config from YAML file and optionlly override some values.
merge_with_args(args)Merge config with command-line arguments.
merge_with_config(config)Merge other config into this config.
to_dictconfig()Convert config to DictConfig.
to_file(path)Write config to YAML file.
to_string()Generate YAML string representation of config.
Attributes
train_data_processingvalid_data_processingtest_data_processing