(遇到问题)(已解决) raise NotImplementedErrorNotImplementedError

1 每次运行到这里都出错，下面是我根据CBAM改成的F3Net的形式

class Flatten(nn.Module):
    def __init__(self, ):
        super(Flatten, self).__init__()

    def forword(self, x):
        return x.view(x.size(0), -1)  # -1表示不确定数

    def initialize(self):
        weight_init(self)

2 这是CBAM原版

class Flatten(nn.Module):
    def forword(self, x):
        return x.view(x.size(0), -1)

3 直接x = x.view(x.size(0), -1)解决了