Search Unity

Disable training on one agent

Discussion in 'ML-Agents' started by jeffsim, Mar 24, 2020.

  1. jeffsim

    jeffsim

    Joined:
    Feb 3, 2016
    Posts:
    8
    Hello; I'm trying to train a couple of different Agents. For context: I'm playing around with NN for an RPG, where one agent would have a "Tank" brain, another has a "Healer" brain, etc. I've trained the Tank brain to orient on a target and attack it until it is dead. I'd now like to train the "Healer" brain with a different set of observations/rewards. I've set it up so that each Agent has its own brain to train, and I've given them different BehaviorNames.

    Here's my problem: now that the Tank brain is trained, I'd like to train the Priest brain to heal the Tank, where the Tank uses its existing training; ie I don't want to retrain the Tank. However; I can't see how to only train only the Healer; when I run mlagents-learn, both are trained.

    Is it possible to tell ml-agents "train *this* agent, but not *that* one (use the existing brain set in BehaviorParameters.Model)"? Apologies if this has a painfully obvious answer...

    Thanks!
     
  2. celion_unity

    celion_unity

    Joined:
    Jun 12, 2019
    Posts:
    289
  3. jeffsim

    jeffsim

    Joined:
    Feb 3, 2016
    Posts:
    8
    Ah, obvious in retrospect :). Thank you!