Skip to content

Draft: change to_onnx to load models using the SaltCLI

Nikita Ivvan Pond requested to merge npond/salt:allow_onnx_ckpt_after_transfer into main

Description

This MR aims to switch the to_onnx script to loading models using the SaltCLI, rather than from the Model.load_checkpoint function. This is required to ensure everything loads correctly, for example in the current form its difficult to train a model on one server, and then export to onnx on another, as the model stores information such as the norm dict path in the checkpoint, which must ve overwritten by the CLI.

This MR aims to close #45 . It is currently a WIP as there seems to be some difficulty in how we deal with ckpt_path

Review checklist:

  • CI Passing
  • Comments addressed
  • Source branch is up to date with target

Merge request reports