Skip to content

v0.4.0 - multi-gpu and result-by-alias

Latest
Compare
Choose a tag to compare
@MiWeiss MiWeiss released this 16 Feb 12:11
· 1 commit to main since this release
e37ea28

What's Changed

  • 🐛 Multi-GPU ensembles are possible again. Just extend the DeviceAllocatorContextManagerV2 context manager.
  • ✨ Add return_alias_dict=True when calling predict_quantified and the return value will be a dict of quantifier aliases pointing to (prediction, confidence_or_uncertainty) tuples.