浏览代码

Fix SAC + LSTM Barracuda inference (#2698)

/develop-gpu-test
GitHub 5 年前
当前提交
9bac2771
共有 1 个文件被更改,包括 5 次插入2 次删除
  1. 7
      ml-agents/mlagents/trainers/sac/models.py

7
ml-agents/mlagents/trainers/sac/models.py


)
# We assume m_size is divisible by 4
# Create the non-Policy inputs
# Use a default placeholder here so nothing has to be provided during
# Barracuda inference. Note that the default value is just the tiled input
# for the policy, which is thrown away.
self.other_memory_in = tf.placeholder(
self.other_memory_in = tf.placeholder_with_default(
input=tf.tile(self.inference_memory_in, [1, 3]),
dtype=tf.float32,
name="other_recurrent_in",
)

正在加载...
取消
保存