浏览代码

Scale size of CC network

/comms-grad
Ervin Teng 4 年前
当前提交
e4e0e2bd
共有 1 个文件被更改,包括 2 次插入1 次删除
  1. 3
      ml-agents/mlagents/trainers/torch/networks.py

3
ml-agents/mlagents/trainers/torch/networks.py


super().__init__()
self.normalize = network_settings.normalize
self.use_lstm = network_settings.memory is not None
self.h_size = network_settings.hidden_units
# Scale network depending on num agents
self.h_size = network_settings.hidden_units * num_obs_heads
self.m_size = (
network_settings.memory.memory_size
if network_settings.memory is not None

正在加载...
取消
保存