edflow.hooks.checkpoint_hooks.lambda_checkpoint_hook module

Summary

Classes:

LambdaCheckpointHook

Reference

class edflow.hooks.checkpoint_hooks.lambda_checkpoint_hook.LambdaCheckpointHook(root_path, global_step_getter, global_step_setter, save, restore, interval=None, ckpt_zero=False, modelname='model')

Bases: edflow.hooks.hook.Hook

__init__(root_path, global_step_getter, global_step_setter, save, restore, interval=None, ckpt_zero=False, modelname='model')

before_epoch(epoch)

Parameters

epoch –

after_epoch(epoch)

Parameters

epoch –

after_step(step, last_results)

Parameters

• step –

• last_results –

at_exception(*args, **kwargs)

Parameters

• *args –

• **kwargs –

save(force_active=False)

static parse_global_step(checkpoint)

Parameters

checkpoint –