api的训练文件中需要
eval_hook = DistEvalHook if distributed else EvalHook
runner.register_hook(
eval_hook(val_dataloader, save_best='bbox_mAP', **eval_cfg), priority='LOW')
将evalhook通过register_hook注册进入钩子列表里,实例化时加入参数save_best=就可,需要什么具体评价指标,就赋什么指标名称 例如'bbox_mAP',类型必须为字符串。
注意:evalhook虽然命名为hook,但是没有使用HOOK.registry_module()进行注册表注册 所以也就不需要bulid。
如何保存save_best具体参见 DistEvalHook
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)