register_hook
接口原型
register_hook(model, config, hook_type="dump_data")
功能说明
给模型添加hook，用于转储数据。
参数说明
参数名 |
说明 |
是否必选 |
|---|---|---|
model |
需要hook的模型，数据类型：torch.nn.Module，建议设置为最外层的torch模型。 |
是 |
config |
hook配置，数据类型：DumpConfig。 |
是 |
hook_type |
hook类型，数据类型：str，默认值为dump_data，当前仅支持dump_data。 |
否 |
父主题： API参考