Method init_hidden

AutoML/darts-master/rnn/model.py:157–159

(self, bsz)

        return model_output, hidden

    def init_hidden(self, bsz):
      weight = next(self.parameters()).data
      return [Variable(weight.new(1, bsz, self.nhid).zero_())]