Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • Peizhao Zhang's avatar
    Added hooks to report training progress to fblearner and keep alive. · bd6043ee
    Peizhao Zhang authored 
    Summary:
* Added a registry to register functions that could be used to register hooks for training.
  * TRAINER_HOOKS_REGISTRY: List of functions to add hooks for trainer, all functions in the registry will be called to add hooks
  * `func(hooks: List[HookBase]) -> None`

Reviewed By: zhanghang1989

Differential Revision: D27560806

fbshipit-source-id: fcfa02623bfd08508b6083db2d318d08f7e3c0b8
    bd6043ee
test_runner_default_runner.py 14.2 KB