Search Unity

  1. Megacity Metro Demo now available. Download now.
    Dismiss Notice
  2. Unity support for visionOS is now available. Learn more in our blog post.
    Dismiss Notice

Already trained model and rewards

Discussion in 'ML-Agents' started by antreas20197, Feb 23, 2022.

  1. antreas20197

    antreas20197

    Joined:
    Jun 9, 2021
    Posts:
    15
    Hello I have a quenstion about rewards and a trained model.

    Is a trained model, with a brain attached to it, uses the rewards (AddReward() function) from training script or rewards are used only in training process?

    I mean, if I attach the brain and remove all AddReward() calls, will the model still work?

    Thanks,
     
  2. mbaske

    mbaske

    Joined:
    Dec 31, 2017
    Posts:
    473
    It'll work - rewards are only relevant for training. You can safely remove them at inference (although keeping them in won't cause any issues either way).
     
  3. antreas20197

    antreas20197

    Joined:
    Jun 9, 2021
    Posts:
    15
    Nice!
    Thanks mbaske for the answer.