class Extractor(nn.Module):

    def __init__(self):
        super(Extractor, self).__init__()

        self.mlp = nn.Linear(768, 1024),
        self.flatten = nn.Flatten()

    def forward(self, x):

        x = self.mlp(x)
        x = self.flatten(x)
        return x

跑深度学习代码的时候突然总报这个错误,后来才发现是代码后面不小心加了逗号。

把逗号去掉就可以了。

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐