【发布时间】：2020-02-05 05:26:22
【问题描述】：
我正在尝试跟踪我保存的重量，以便可以使用 Catalyst 加载它。我检查了 PyTorch 上的文档，但我只发现了函数的跟踪。
【问题讨论】：
-
什么是“追踪重量”？什么是“催化剂”？这不是一个常见的术语。
标签： python machine-learning pytorch
【发布时间】：2020-02-05 05:26:22
【问题描述】：
您可以跟踪torch.nn.Module 实例。
如果您不使用数据相关流控制，只需使用torch.jit.trace，例如：
torch.jit.trace(module, example_input)
如果是，请使用torch.jit.script:
torch.jit.script(module)
您可以找到更多信息inside tutorials 或相应的文档。
【讨论】：