创建模块时的AttributeError

时间:2017-06-15 21:43:30

标签: pytorch

我收到以下错误。

  

AttributeError:无法在Module。 init ()调用

之前分配模块

我有一个课程如下。

class Classifier(nn.Module):

    def __init__(self, dictionary, embeddings_index, max_seq_length, args):  
        self.embedding = EmbeddingLayer(len(dictionary), args.emsize, args.dropout)
        self.drop = nn.Dropout(args.dropout)

我在这里做错了什么?我是PyTorch的初学者,请帮忙!

1 个答案:

答案 0 :(得分:2)

创建模块时,您应该始终做的第一件事就是调用它的超级构造函数。所以,你的课应该是这样的:

class Classifier(nn.Module):

    def __init__(self, dictionary, embeddings_index, max_seq_length, args): 
        super(Classifier, self).__init__() 
        '''Rest of your code goes here.'''