训练包含多个节点的控制策略

控制任务中控制多个节点实现目标是常见的需求。Revive SDK支持通过简单的配置训练包含多个节点的控制策略。

例如,在自动驾驶任务中,需要同时控制车辆的方向和动力的输出。这两个方面都对车辆的行驶方向产生影响,因此需要同时考虑它们。通过使用 Revive SDK。

下面通过这个示例展示如何在训练时启动该功能:

metadata:

   graph:
     action_1:
     - observation
     action_2:
     - action_1
     - observation
     next_observation:
     - action_1
     - action_2
     - observation

   columns:
   ...

上面的决策流图,展示了一个多节点的控制业务, action_1 节点和 action_2 节点需要共同作用完成控制。在训练时使用 -tpn 参数进行多策略节点训练。

训练命令:

python train.py -df test.npz -cf test.yaml -rf test_reward.py -vm once -pm once -tpn action_1,action_2 --run_id test