Search Unity

  1. Get the latest news, tutorials and offers directly to your inbox with our newsletters. Sign up now.
    Dismiss Notice

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:
    9
    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.
     
  2. andrewcoh_unity

    andrewcoh_unity

    Unity Technologies

    Joined:
    Sep 5, 2019
    Posts:
    143
    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.
     
    prismspecsx likes this.
unityunity