Skip to content

TrainerConfig

Base configuration for GRADIEND trainers.

Holds parameters that are independent of modality (text, image, etc.), such as visualization/plotting options used by all trainers.

img_dpi class-attribute instance-attribute

img_dpi = None

DPI for saved plots (e.g. 600 for publication). None = use visualizer default.

img_format class-attribute instance-attribute

img_format = 'png'

Image format for plots (e.g. encoder distributions, training convergence). Forwarded to visualizer.

__str__

__str__()
Source code in gradiend/trainer/config.py
def __str__(self) -> str:
    return f"TrainerConfig(img_format={self.img_format!r}, img_dpi={self.img_dpi})"