Python TypeError - 'Class' 对象不可调用(Google 内部协作示例)

Python TypeError - 'Class' object is not callable (Google Collab Example Inside)

[已编辑]

在这个例子中,在我最后的代码单元格中,我尝试调用我的模型。这是按照 Youtube 视频上的教程进行的。

在这一步,视频能够执行台词

model = UCC_Classifier(config)

然后在下一个单元格

loss, output = model(input_ids.unsqueeze(dim=0), am.unsqueeze(dim=0), labels.unsqueeze(dim=0))

成功得到结果。然而,当我尝试做同样的事情时,我被告知我的 class 不可调用。我看不出有什么区别,也不确定为什么这可能无法调用。

谢谢

您的 UCC_Classifier 模型应该是 pl.LightningModule,而不是 pl.LightningDataModule