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_processing

valid_data_processing

test_data_processing