search

kennymckormick / pyskl

A toolbox for skeleton-based action recognition.
Apache License 2.0
911 stars 175 forks source link

Training register custom hooks #212

Open ggggg111 opened 8 months ago

ggggg111 commented 8 months ago

At the current state, custom hooks from config files are not registered. E.g.:

cfg.custom_hooks = [
    dict(
        type='CustomHook1',
        priority='HIGHEST',
        # args [...]
    )
]

This can be easily solved by modifying the current train_model function after registering the standard hooks (see https://github.com/kennymckormick/pyskl/blob/main/pyskl/apis/train.py#L117).