火炬跟踪-如何跟踪节省的重量？

Question

我正在尝试跟踪所节省的重量，以便可以在Catalyst中加载它。我查看了PyTorch上的文档，但只发现了对函数的跟踪。

Answer 1

您可以跟踪torch.nn.Module个实例。

如果您不使用依赖数据的流控制，则只需使用torch.jit.trace，例如：

torch.jit.trace(module, example_input)

如果有，请使用torch.jit.script：

torch.jit.script(module)

您可以找到更多信息inside tutorials或相应的文档。