浏览代码

Barracuda hotfix for TF2 AddV2 support (#2703)

/develop-gpu-test
Chris Elion 5 年前
当前提交
1eee5b5c
共有 1 个文件被更改,包括 1 次插入0 次删除
  1. 1
      ml-agents/mlagents/trainers/tensorflow_to_barracuda.py

1
ml-agents/mlagents/trainers/tensorflow_to_barracuda.py


"OneHot": Struct(id=67, rank=lambda inputs: inputs[0] + 1),
# Broadcast ops
"Add": Struct(id=100, rank=lambda inputs: np.max(inputs)),
"AddV2": Struct(id=100, rank=lambda inputs: np.max(inputs)),
"Sub": Struct(id=101, rank=lambda inputs: np.max(inputs)),
"Mul": Struct(id=102, rank=lambda inputs: np.max(inputs)),
"RealDiv": Struct(id=103, rank=lambda inputs: np.max(inputs)),

正在加载...
取消
保存