Search Unity

  1. Unity support for visionOS is now available. Learn more in our blog post.
    Dismiss Notice

Display the policy observation vectors weights

Discussion in 'ML-Agents' started by Jean-Diddy, Jul 14, 2020.

  1. Jean-Diddy

    Jean-Diddy

    Joined:
    Jun 29, 2020
    Posts:
    4
    Hello,

    In order to choose the select the observations vectors which are necessary to fast the training of the agent, I would like to display the policy weights of the PPO trainer.

    Is there a way to do so ? I looked at the python / tensorflow related code but it is really hard for me to know how to change the code so as to do so...

    Thank you !
     
  2. vincentpierre

    vincentpierre

    Joined:
    May 5, 2017
    Posts:
    160
    Hi, unfortunately, we do not offer this functionality as of today. I would recommend loading the frozen_graph_def.pb located in the results folder using TensorFlow and inspect the weights directly inside it.