Search Unity

how to load an existing TF model in Unity

Discussion in 'ML-Agents' started by prismspecsx, Apr 12, 2021.

  1. prismspecsx

    prismspecsx

    Joined:
    Nov 27, 2018
    Posts:
    13
    Are there any docs on this? I have an existing .pb that does some math for me which I would love to access inside of Unity. Not agent based stuff, just general ML.
     
    prismspecsx, Apr 12, 2021
    #1

  2. andrewcoh_unity

    andrewcoh_unity

    Unity Technologies

    Joined:
    Sep 5, 2019
    Posts:
    162
    Hi @prismspecsx

    This sounds like a more general Barracuda question (forum and github) which is the inference engine used to run NNs inside of Unity. Also, you'll likely have to convert the .pb to .nn or .onnx.

    ML-Agents can really only guarantee usage of models generated by ML-Agents and our model runner pipeline (code) is tightly integrated with the ML Agents loop.
     
    andrewcoh_unity, Apr 12, 2021
    #2
    prismspecsx likes this.