【 在 feng321 的大作中提到: 】
: [code=py]
: class WordEmbeddingWrapper(torch.nn.Module):
: def __init__( self, original_word_embedding):
: ...................
># 使用torch.jit.trace对模型进行跟踪编译
> traced_model = torch.jit.trace(word_embedding_wrapper, example_input)
你想想, 如果不做trace跟踪该怎么写
是不是这样:
word_embedding_wrapper(example_input)
实际上调用的是类的__call__方法。
word_embedding_wrapper.__call__(example_input)
现在,只不过是被trace包装了一下:
--
FROM 124.126.1.*