pyskl icon indicating copy to clipboard operation
pyskl copied to clipboard

Training register custom hooks

Open ggggg111 opened this issue 2 years ago • 0 comments

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).

ggggg111 avatar Nov 20 '23 06:11 ggggg111