浏览代码

Move some functionality to optimizer-black

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

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


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

正在加载...
取消
保存