We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent ada3bc5 commit d41a2a0Copy full SHA for d41a2a0
ml-agents/mlagents/trainers/sac/models.py
@@ -556,10 +556,13 @@ def create_memory_ins(self, m_size):
556
)
557
# We assume m_size is divisible by 4
558
# Create the non-Policy inputs
559
+ # Use a default placeholder here so nothing has to be provided during
560
+ # Barracuda inference. Note that the default value is just the tiled input
561
+ # for the policy, which is thrown away.
562
three_fourths_m_size = m_size * 3 // 4
- self.other_memory_in = tf.placeholder(
563
+ self.other_memory_in = tf.placeholder_with_default(
564
+ input=tf.tile(self.inference_memory_in, [1, 3]),
565
shape=[None, three_fourths_m_size],
- dtype=tf.float32,
566
name="other_recurrent_in",
567
568
0 commit comments