Language_Model.py 文件源码

def __init__(self, ninput, nembed, nhidden, nlayers, bias, dropout):
        super(Img_Encoder_Structure, self).__init__()
        self.image_encoder = FC(ninput, nembed, relu=True)
        self.rnn = nn.LSTM(nembed, nhidden, nlayers, bias=bias, dropout=dropout)