Search Unity

Export the neural network model every 'n' number of steps

Discussion in 'ML-Agents' started by blockimperium, Oct 26, 2020.

  1. blockimperium

    blockimperium

    Joined:
    Jan 21, 2008
    Posts:
    452
    One of the thing that I've been trying to find an option for in the release version of ML Agents (7 to 8) is a way to tell mlagents that I want it to routinely have it dump out the model ever 'n' number of steps. This is useful in being able to see the capability of the agent after it trains at regular intervals.

    Based on some older blog posts it seems there was an option for it in the past, but there doesn't appear to be a flag for it anymore.
     
  2. celion_unity

    celion_unity

    Joined:
    Jun 12, 2019
    Posts:
    289
    The "checkpoint_interval" field in your config file determines how often we save checkpoints (it defaults to 500000). Since Release 6, saving checkpoints also creates the .nn and/or .onnx files.
     
    blockimperium likes this.