Skip to content

1.2.5

Latest
Compare
Choose a tag to compare
@marovira marovira released this 13 Feb 02:18
· 11 commits to master since this release

Feature Changes

  • helios.losses.LOSS_REGISTRY now ships with all built-in PyTorch losses registered.
  • helios.onnx.export_to_onnx can now delete the exported network if validation fails.
  • helios.data now ships with a COLLATE_FN_REGISTRY that can be used for custom collate functions.
  • All dependencies have been updated.

Bug fixes

  • Exporting an ONNX network larger than 2GB no longer causes a crash.
  • helios.scheduler.create_scheduler and helios.optim.create_optimizer now have correct type-hints.

Note

Helios does not currently support Python 3.13 as onnx doesn't support it yet.