我的nn.Module类的 __call__方法,是这样的
def __call__(self, *args: Any, **kwargs: Any) -> Any:
if self.with_module:
module = self.module()
if module is None:
raise RuntimeError("You are trying to call the hook of a dead Module!")
return self.hook(module, *args, **kwargs)
return self.hook(*args, **kwargs)
这是如何对forword方法的调用的呢?谢谢
【 在 hgoldfish 的大作中提到: 】
: 因为 nn.Module 有个 __call__() 方法,所以被实例化以后,可以当作函数来用。
:
--
FROM 114.99.170.*