浏览代码

update type of evaluate_actions to list tensor (#4747)

/MLA-1734-demo-provider
GitHub 4 年前
当前提交
8ab2e619
共有 1 个文件被更改,包括 2 次插入2 次删除
  1. 4
      ml-agents/mlagents/trainers/policy/torch_policy.py

4
ml-agents/mlagents/trainers/policy/torch_policy.py


def evaluate_actions(
self,
vec_obs: torch.Tensor,
vis_obs: torch.Tensor,
vec_obs: List[torch.Tensor],
vis_obs: List[torch.Tensor],
actions: AgentAction,
masks: Optional[torch.Tensor] = None,
memories: Optional[torch.Tensor] = None,

正在加载...
取消
保存