Search Unity

Ml-agents low level python api: How do I activate training mode? Is there still something like that?

Discussion in 'ML-Agents' started by schmiJo1, Jun 1, 2021.

  1. schmiJo1

    schmiJo1

    Joined:
    May 20, 2021
    Posts:
    3
    Hello,
    I am used to an earlier version of unity agents, where there was a an train_mode option in the reset function of the environment:
    env.reset(train_mode=True)[brain_name]
    I have not found anything similar in the documentation for ml-agents.

    The only thing I found that goes into this direction is the side-channels option, but I don't know how to configure them to optimally train my model.

    Help is greatly appreciated.
    Thank you
     
  2. vincentpierre

    vincentpierre

    Joined:
    May 5, 2017
    Posts:
    160
    Hi,
    train_mode was removed because it was a confusing name. train_mode did not mean that there would be training, it only meant that the engine would try to simulate faster (higher timescale and set captureframerate). You are right, it was replaced with the engine_configuration_side_channel on Python.