How do I select/swap networks at run time?

I’m working on a network that teaches a quadruped to roll over.
in a separate network, the quadruped knows how to walk towards a goal.

I’d like to run these from a game and swap them in the right context.

How, please?

I’ve done that for changing difficulty level for my game setting. Switching NN models on runtime can be implemented by using SetModel method as shown in the sample code below:

using Unity.MLAgents;
using Unity.MLAgents.Policies;
using Unity.Barracuda;

// Difficulty Levels
public enum BattleMode
{
   Easy,
   Medium,
   Hard
}

// Unity.Barracuda.NNModel class
public NNModel easyBattleBrain = "BossBattle-easy.onnx";
public NNModel mediumBattleBrain = "BossBattle-medium.onnx";
public NNModel hardBattleBrain = "BossBattle-hard.onnx";

// Should match with Agent's Behavior Name in Unity's UI
public string behaviorName = "BossBattle";

public void setDifficulty(BattleMode mode, Agent agent)
{
   switch (mode)
   {
       case BattleMode.Easy:
           agent.SetModel(behaviorName, easyBattleBrain);
           break;
       case BattleMode.Medium:
           agent.SetModel(behaviorName, mediumBattleBrain);
           break;
       case BattleMode.Hard:
           agent.SetModel(behaviorName, hardBattleBrain);
           break;
       default:
           break;
   }
}