def _pack_runner(runner: IRunner): checkpoint = utils.pack_checkpoint( model=runner.model, criterion=runner.criterion, optimizer=runner.optimizer, scheduler=runner.scheduler, epoch_metrics=dict(runner.epoch_metrics), valid_metrics=dict(runner.valid_metrics), stage_name=runner.stage_name, epoch=runner.epoch, loader_name=runner.loader_name, loader_step=runner.loader_batch_step, global_epoch=runner.global_epoch, checkpoint_data=runner.checkpoint_data, main_metric=runner.main_metric, minimize_metric=runner.minimize_metric, valid_loader=runner.valid_loader, ) return checkpoint
def _pack_state(state: State): checkpoint = utils.pack_checkpoint( model=state.model, criterion=state.criterion, optimizer=state.optimizer, scheduler=state.scheduler, epoch_metrics=dict(state.epoch_metrics), valid_metrics=dict(state.valid_metrics), stage_name=state.stage_name, epoch=state.epoch, loader_name=state.loader_name, loader_step=state.loader_batch_step, global_epoch=state.global_epoch, checkpoint_data=state.checkpoint_data, main_metric=state.main_metric, minimize_metric=state.minimize_metric, valid_loader=state.valid_loader, ) return checkpoint