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