Skip to content

Update ONNX export

Dmitrii Kobylianskiirequested to merge
dkobylia/salt:dkobylia/onnx-export-update into main

Summary

This MR introduces the following changes

  • Add support for GN3 export
  • Add test pipelines for GN3V00 and GN3V01 versions
  • Fix issue with non-correct weight handling in the loss functions. We tried to pass list[float] instead of torch.Tensor when loaded from checkpoint.
  • Update type hints for check.py

Conformity

Edited by Dmitrii Kobylianskii

Merge request reports