Search Unity

Question mlagents not generating nn file

Discussion in 'ML-Agents' started by AngrySamsquanch, Dec 1, 2020.

  1. AngrySamsquanch

    AngrySamsquanch

    Joined:
    Dec 28, 2014
    Posts:
    24
    mlagents is not generating a nn file at the end of training. Is there a way to create one from a checkpoint or an onnx file?
     
  2. celion_unity

    celion_unity

    Joined:
    Jun 12, 2019
    Posts:
    289
    As of Release 10, the default behavior is to use pytorch for trainers, which only produce .onnx files. You can still force tensorflow for training (which will produce .nn files, and .onnx files if you have tf2onnx installed) by passing "--tensorflow" on the commandline to mlagents-learn, or setting "framework: tensorflow" in the configuration file. Note that we'll likely be removing tensorflow support in the next release or two.

    There is no way to create a .nn file from .onnx files or checkpoints, and we do not plan to add one.

    Can I ask why you don't want to use the .onnx file?
     
  3. AngrySamsquanch

    AngrySamsquanch

    Joined:
    Dec 28, 2014
    Posts:
    24
    I didn't realize that you could. A lot of the tutorials and documentation seems to focus on the older tf implementation. Thanks for the help.
     
  4. celion_unity

    celion_unity

    Joined:
    Jun 12, 2019
    Posts:
    289
    I think we're a little behind about updating the examples and documentation. Not much we can do about the existing tutorials, though. But they should behave the same.