Python PyTorch PackageExporter.register_intern_hook用法及代碼示例

本文簡要介紹python語言中 torch.package.PackageExporter.register_intern_hook 的用法。

用法:

register_intern_hook(hook)

返回：

一個句柄，可用於通過調用 handle.remove() 來刪除添加的鉤子。

返回類型：

torch.utils.hooks.RemovableHandle

在導出器上注冊一個實習生掛鉤。

每次模塊與 intern() 模式匹配時都會調用該鉤子。它應該具有以下簽名：

hook(exporter: PackageExporter, module_name: str) -> None

Hooks 將按照注冊的順序被調用。