浏览代码

fix tests

/develop/singular-embeddings
vincentpierre 4 年前
当前提交
7f8e6a0d
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 2
      ml-agents/mlagents/trainers/tests/torch/test_utils.py

2
ml-agents/mlagents/trainers/tests/torch/test_utils.py


obs_shapes.append(vis_obs_shape)
h_size = 128
sen_spec = create_sensor_specs_with_shapes(obs_shapes)
encoders, embedding_sizes = ModelUtils.create_input_processors(
encoders, embedding_sizes, _ = ModelUtils.create_input_processors(
sen_spec, h_size, encoder_type, normalize
)
total_output = sum(embedding_sizes)

正在加载...
取消
保存