ncalab.training.trainingparameters

Classes

TrainingParameters

Module Contents

class ncalab.training.trainingparameters.TrainingParameters(gradient_clipping: bool = False, lr: float | None = None, lr_gamma: float = 0.9999, adam_betas: tuple = (0.9, 0.99), batch_repeat: int = 2, max_epochs: int = 200, optimizer_method: str = 'adamw')

_gradient_clipping = False

_lr_gamma = 0.9999

_adam_betas = (0.9, 0.99)

info() → str

Shows a markdown-formatted info string with training parameters. Useful for showing info on tensorboard to keep track of parameter changes.

Returns [str]:

Markdown-formatted info string.

save(path)