Search Unity

Question easy way to do pruning?

Discussion in 'ML-Agents' started by amplez, Oct 19, 2022.

  1. amplez

    amplez

    Joined:
    Aug 19, 2018
    Posts:
    5
    I'm trying to get the size of my agents down and after some reading i believe it can be done with pruning.
    I am not very familiar with pytorch and most of my work on ml was through the mlagents package.

    how should i go about pruning the agent neural network? any advise is appreciated before I delve in to the subject.