Skip to content

Modify to_onnx script to load correct norm dict

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

Description

This MR directly passed the norm_dict defined in the relevant salt config when loading a model before converting it to onnx. This closes issue #45 (closed) , allowing the user to now run to_onnx on a different system than the the model was trained on.

Review checklist:

  • CI Passing
  • Comments addressed
  • Source branch is up to date with target
Edited by Nikita Ivvan Pond

Merge request reports