浏览代码

[bug-fix] Fix error with discrete probs (#4309)

/develop/add-fire
GitHub 4 年前
当前提交
d8db1477
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 2
      ml-agents/mlagents/trainers/torch/utils.py

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


entropies = entropies.squeeze(-1)
all_probs = None
else:
all_probs = torch.cat(all_probs, dim=-1)
all_probs = torch.cat(all_probs_list, dim=-1)
return log_probs, entropies, all_probs
正在加载...
取消
保存