浏览代码

Renaming GOAL to GOAL_SIGNAL

/goal-conditioning/sensors-3-pytest-fix
vincentpierre 4 年前
当前提交
1b4fd8fb
共有 2 个文件被更改,包括 2 次插入2 次删除
  1. 2
      ml-agents/mlagents/trainers/tests/simple_test_envs.py
  2. 2
      ml-agents/mlagents/trainers/torch/networks.py

2
ml-agents/mlagents/trainers/tests/simple_test_envs.py


obs_spec[i] = ObservationSpec(
shape=obs_spec[i].shape,
dimension_property=obs_spec[i].dimension_property,
observation_type=ObservationType.GOAL,
observation_type=ObservationType.GOAL_SIGNAL,
name=obs_spec[i].name,
)
return obs_spec

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


self._total_goal_enc_size = 0
self._goal_processor_indices: List[int] = []
for i in range(len(observation_specs)):
if observation_specs[i].observation_type == ObservationType.GOAL:
if observation_specs[i].observation_type == ObservationType.GOAL_SIGNAL:
self._total_goal_enc_size += self.embedding_sizes[i]
self._goal_processor_indices.append(i)

正在加载...
取消
保存