espnet2.train.deepspeed_trainer.DeepSpeedTrainerOptions

source

class espnet2.train.deepspeed_trainer.DeepSpeedTrainerOptions(resume: bool, seed: int, train_dtype: str | torch.dtype, log_interval: int | None, output_dir: pathlib.Path | str, max_epoch: int, deepspeed_config: pathlib.Path | str)

Bases: object

deepspeed_config : Path | str

log_interval : int | None

max_epoch : int

output_dir : Path | str

resume : bool

seed : int

train_dtype : str | dtype