在写引用model class的时候，发现无法算出prediction。报错如下

• 报错：

  • in forward raise NotImplementedError

• 解决：

  • 将forward的两个下划线去掉即可，疑似版本问题

  • 报错时是

    def __forward__(self, xb):
        return xb.float() @ self.weights + self.bias
  

  • 不报错时是

    def forward(self, xb):
        return xb.float() @ self.weights + self.bias
  

Posted in Code with : 图像处理, ML