TrainConfig#

class TrainConfig(
do=True,
precision='32-true',
iterations=-1,
batch_size=-1,
grad_clip=0.0,
track_grad_norm=-1,
accumulate_grad_steps=1,
run_start_time=None,
run_time_limit_hours=None,
)#

Bases: object

Train configuration.

Parameters:

  • do (bool)

  • precision (str)

  • iterations (int)

  • batch_size (int)

  • grad_clip (float)

  • track_grad_norm (int)

  • accumulate_grad_steps (int)

  • run_start_time (float | None)

  • run_time_limit_hours (float | None)

do: bool = True#
precision: str = '32-true'#
iterations: int = -1#
batch_size: int = -1#
grad_clip: float = 0.0#
track_grad_norm: int = -1#
accumulate_grad_steps: int = 1#
run_start_time: float | None = None#
run_time_limit_hours: float | None = None#
__init__(
do=True,
precision='32-true',
iterations=-1,
batch_size=-1,
grad_clip=0.0,
track_grad_norm=-1,
accumulate_grad_steps=1,
run_start_time=None,
run_time_limit_hours=None,
)#
Parameters:

  • do (bool)

  • precision (str)

  • iterations (int)

  • batch_size (int)

  • grad_clip (float)

  • track_grad_norm (int)

  • accumulate_grad_steps (int)

  • run_start_time (float | None)

  • run_time_limit_hours (float | None)

Return type:

None