浏览代码

removing extra line

/exp-tanh
vincentpierre 4 年前
当前提交
099a51ef
共有 1 个文件被更改,包括 0 次插入1 次删除
  1. 1
      ml-agents/mlagents/trainers/torch/networks.py

1
ml-agents/mlagents/trainers/torch/networks.py


encoding = encoding.reshape([-1, sequence_length, self.h_size])
encoding, memories = self.lstm(encoding, memories)
encoding = encoding.reshape([-1, self.m_size // 2])
return encoding, memories

正在加载...
取消
保存