TrainerState

class braintools.trainer.TrainerState

Container for trainer state during training.

epoch

Current epoch.

Type:

int

global_step

Total number of training steps.

Type:

int

stage

Current stage (‘train’, ‘validate’, ‘test’, ‘predict’).

Type:

str