Search Unity

Question How to use graph neural network in ml-agents?

Discussion in 'ML-Agents' started by MingZiJiaoZhongGuo, Dec 11, 2023.

  1. MingZiJiaoZhongGuo

    MingZiJiaoZhongGuo

    Joined:
    Apr 3, 2021
    Posts:
    4
    I plan to use a graph neural network to extract scene-related features during the training process of my reinforcement learning model with Unity. However, I couldn't find an existing implementation for this. I came across a frequently used PyTorch library called PyG and I'm wondering how to combine it with Unity's ml-agent. Can someone please guide me on how to do it?
     

    Attached Files:

    • .jpg
      .jpg
      File size:
      51.6 KB
      Views:
      28