浏览代码

Move some functionality to optimizer-black

/develop/nopreviousactions
Ervin Teng 5 年前
当前提交
2c1ef594
共有 1 个文件被更改,包括 7 次插入2 次删除
  1. 9
      ml-agents/mlagents/trainers/ppo/optimizer.py

9
ml-agents/mlagents/trainers/ppo/optimizer.py


m_size=None,
seed=0,
stream_names=None,
vis_encode_type=EncoderType.SIMPLE
vis_encode_type=EncoderType.SIMPLE,
):
"""
Takes a Unity environment and model-specific hyper-parameters and returns the

:param num_layers: Number of hidden linear layers.
"""
hidden_streams = LearningModel.create_observation_streams(
self.policy.visual_in, self.policy.processed_vector_in, 1, h_size, num_layers, vis_encode_type
self.policy.visual_in,
self.policy.processed_vector_in,
1,
h_size,
num_layers,
vis_encode_type,
)
if self.use_recurrent:

正在加载...
取消
保存