peal.generators.interfaces.GeneratorConfig

class peal.generators.interfaces.GeneratorConfig[source]

Bases: BaseModel

This class defines the config of a generator.

Methods

Attributes

batch_size

The batch size of the generator.

category

The category of the config

current_fid

The name of the class.

seed

generator_type

The type of generator that shall be used.

generator_type: str

The type of generator that shall be used.

category: str = 'generator'

The category of the config

batch_size: int = 1

The batch size of the generator.

current_fid: float = inf

The name of the class.

seed: int = 0