浏览代码

Scale value network based on num agents

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

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


if network_settings.memory is not None:
encoding_size = network_settings.memory.memory_size // 2
else:
encoding_size = network_settings.hidden_units
encoding_size = network_settings.hidden_units * num_agents
self.value_heads = ValueHeads(stream_names, encoding_size, outputs_per_stream)
def forward(

正在加载...
取消
保存