Torchscript调用其他函数而不是转发

Question

当我将torch模型编译成torchscript时，只需调用torchscript模型对象forward，就可以使用函数model()。

但是，当我想使用在模型上创建的另一个函数时，我不能调用该函数。我尝试执行model.functionName()，期望调用函数functionName，但什么也没有发生。

对于如何调用不向前的模型对象的子函数，有什么想法吗？

谢谢

Answer 1

我认为您应该在您想要调用的方法之上添加装潢工 @torch.jit.export，并再次转换torchscript模型。之后，您可以通过module.run_method(name, args) 文档调用方法。