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 aCOLLATE_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
andhelios.optim.create_optimizer
now have correct type-hints.
Note
Helios does not currently support Python 3.13 as onnx
doesn't support it yet.