Skip to content

coremltools 5.0b5

Pre-release
Pre-release
Compare
Choose a tag to compare
@TobyRoseman TobyRoseman released this 22 Sep 18:49
· 318 commits to main since this release
30ea1da
  • Added support for pytorch conversion for tensor assignment statements: torch_tensor_assign op and index_put_ op . Fixed bugs in translation of expand ops and sort ops.
  • Model input/output name sanitization: input and output names for "neuralnetwork" backend are sanitized (updated to match regex [a-zA-Z_][a-zA-Z0-9_]*), similar to the "mlprogram" backend. So instead of producing input/output names such as "1" or "input/1", "var_1" or "input_1", names will be produced by the unified converter API.
  • Fixed a bug preventing a Model Package from being saved more than once to the same path.
  • Various bug fixes, optimizations and additional testing.