Skip to content
This repository has been archived by the owner on Sep 18, 2024. It is now read-only.

fix issue #2091 #2107

Merged
merged 6 commits into from
Feb 29, 2020
Merged
Changes from 5 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion src/sdk/pynni/nni/nas/pytorch/enas/mutator.py
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,7 @@ def forward(self, inputs, hidden):
curr_c, curr_h = m(inputs, (prev_c[i], prev_h[i]))
next_c.append(curr_c)
next_h.append(curr_h)
inputs = curr_h[-1]
inputs = curr_h[-1].view(1, -1)
return next_c, next_h


Expand Down