Search Unity

  1. We are migrating the Unity Forums to Unity Discussions. On July 12, the Unity Forums will become read-only. On July 15, Unity Discussions will become read-only until July 18, when the new design and the migrated forum contents will go live. Read our full announcement for more information and let us know if you have any questions.

Model is .nn instead of .onnx, is it ONNX model?

Discussion in 'ML-Agents' started by kyoimo, Jun 3, 2022.

  1. kyoimo

    kyoimo

    Joined:
    Apr 13, 2022
    Posts:
    1
    I am using ML-Agents 1.0.8 and followed the Hummingbird tutorial at https://learn.unity.com/course/ml-agents-hummingbirds?uv=2019.3. My trained model is .nn instead of .onnx. I read that it should be .onnx since ML-Agents uses Pytorch. Has this changed? I can't find .onnx file anywhere in my directory. Also, I viewed the .nn model in https://netron.app/, it seems that the documentation is not supported for the model. That is, if I clicked a node in the graph, and clicked ? in "Documentation", nothing shows up.

    I am asking because I want to convert .onnx file for use outside of Unity (e.g., a real robot). Now, I am confused what format my model is in.