How to use custom model in Learning Agents?

No sorry. It’s not on the current roadmap. Here’s the tentative 5.6 roadmap:

  • Training Scale Out
    • Multi-Processed Training: We should be able to easily spawn X number of Unreal processes and have them all communicating with a single python process to be able to improve training throughput.
  • :white_check_mark: Easy Tensorboard Installation
    • Simply use the new “Tensorboard” plugin - no longer need to find the python env and manually install it via pip
  • Imitation Learning 2.0 - Offline RL
    • Implement Soft Actor Critic / Offline Actor Critic
  • Unreal Editor Integration (Stretch Goal)
    • We need this in support of the above “training scale out” goal

I need these things for myself so they are my top priority.

You can patch in custom models yourself if you’re brave enough :slight_smile: I would consider merging a PR if someone put a lot of love into the work. Here’s the thing though: I don’t think custom ONNX models are a good idea… there be dragons.

Brendan