浏览代码

Fix for discrete control image observations

/develop-generalizationTraining-TrainerController
Arthur Juliani 7 年前
当前提交
6ad7f010
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 2
      python/trainers/ppo_models.py

2
python/trainers/ppo_models.py


bw = brain.camera_resolutions[i]['blackAndWhite']
visual_encoders.append(
self.create_visual_encoder(height_size, width_size, bw, h_size, 1, tf.nn.elu, num_layers)[0])
hidden_visual = [tf.concat(visual_encoders, axis=1)]
hidden_visual = tf.concat(visual_encoders, axis=1)
if brain.state_space_size > 0:
s_size = brain.state_space_size * brain.stacked_states
if brain.state_space_type == "continuous":

正在加载...
取消
保存