浏览代码

Fix hypernetwork

/goal-conditioning/new
Arthur Juliani 3 年前
当前提交
fc1ee931
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 2
      ml-agents/mlagents/trainers/torch/networks.py

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


encoded_self = torch.cat([encoded_self, actions], dim=1)
if self.conditioning_type == ConditioningType.DEFAULT:
obs_encodes = obs_encodes + goal_encodes
encoded_self = torch.cat([encoded_self, torch.cat(goal_encodes, dim=-1)], dim=1)
goal_encodes = []
if len(obs_encodes) == 0:

正在加载...
取消
保存