ertk.pytorch.train.PyTorchLoggingConfig

class ertk.pytorch.train.PyTorchLoggingConfig(log_dir: str | None = '', tensorboard: bool = True, csv: bool = True)

Bases: ERTKConfig

__init__(log_dir: str | None = '', tensorboard: bool = True, csv: bool = True) None

Methods

__init__([log_dir, tensorboard, csv])

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

csv

log_dir

tensorboard