What's Changed
- Temporary pinned mlflow version to fix the failing CI by @vfdev-5 in #3278
- Add rank correlation metrics by @kzkadc in #3276
- Helper function
all_gather_tensors_with_shapes()by @sadra-barikbin in #3281 - Fix mypy issue with wandb.finish() by @vfdev-5 in #3283
- To fix #3277 by @puhuk in #3280
- Modify docstring and argument for rank correlation metrics by @kzkadc in #3284
- Use the default runner type by @ZainRizvi in #3285
- [CI] Fixed failing linkcheck, w&b init by @vfdev-5 in #3287
- add HSIC metric by @kzkadc in #3282
- Drop python 3.8 in the CI by @vfdev-5 in #3288
- Fix pytorch versions ci by @vfdev-5 in #3289
- Modify docstring for rank correlation metrics by @kzkadc in #3291
- simplify imports of metric functions by @kzkadc in #3292
- Update import style for PrecisionRecallCurve metric by @kzkadc in #3293
- Fixed mypy failures on the CI by @vfdev-5 in #3296
- Updated code formatting tools' versions by @vfdev-5 in #3297
- Add clustering metrics by @kzkadc in #3290
- Updated pytorch version in CI and docker by @vfdev-5 in #3298
- Fixed torch version in docker push_all.sh by @vfdev-5 in #3300
- Fixed failing tests on pytorch nightly using torch.load by @vfdev-5 in #3299
- try updating gpu tests GHA by @vfdev-5 in #3306
- Update pyproject.toml by @vfdev-5 in #3305
- Fix pynvml version to <12 by @vfdev-5 in #3310
- Fixed GPU tests exec scripts and failing metrics by @vfdev-5 in #3301
- Updated GpuInfo metric, pynvml<12 by @vfdev-5 in #3311
- Give the option to terminate the engine without firing Events.COMPLET… by @bonassifabio in #3309
- Allow to terminate an epoch without firing
Events.EPOCH_COMPLETEDby @bonassifabio in #3313 - Fix deprecated statement by @Gulin7 in #3307
- Fixed failing test_roc_auc.py::test_check_compute_fn test by @vfdev-5 in #3316
- Update pytorch-version-tests.yml by @vfdev-5 in #3315
- Fix typos in TextCNN.jpynb by @k22036 in #3320
- Fix typos in FashionMNIST.jpynb by @k22036 in #3321
- Refactor FashionMNIST notebook to use
next()for data iterator by @k22036 in #3322 - Fixed failing doctest and linkcheck CI jobs by @vfdev-5 in #3331
- Fixed failing test_miou on M1 with pytest.approx by @vfdev-5 in #3330
- Updated pytorch version in docker and pytorch-version-tests.yml by @vfdev-5 in #3329
- Fix a typo in the documentation by @jscanvic in #3332
- COCO mAP metric by @sadra-barikbin in #2901
- Updated accuracy, precision, recall tests by @vfdev-5 in #3333
- Fixed failing mean_average_precision tests on cuda by @vfdev-5 in #3338
- Use f32 for metrics on mps by @vfdev-5 in #3334
- Update test_average_precision.py to resolve issue #3335 by @puhuk in #3337
- Fix group as list of ints in torch dist collective ops by @vfdev-5 in #3340
- Added available_device fixture to test_mean_squared_error by @HyeSungP in #3341
- Add
_rank_not_in_groupto idist by @sadra-barikbin in #3339 - Update test_roc_auc.py to resolve issue #3335 by @puhuk in #3343
- Update test_cosine_similarity.py to resolve issue #3335 by @chaojessica in #3348
- Added available_device in test_classification_report (#3335) by @HyeSungP in #3342
- Run hvd tests without xdist by @vfdev-5 in #3347
- Updated gpu-tests CI by @vfdev-5 in #3349
- adds argument name for device in test_mean_squared_error #3335 by @BanzaiTokyo in #3351
- adds checks for used device in metrics tests. #3335 by @BanzaiTokyo in #3353
- Added available_device in metric tests (#3335) by @HyeSungP in #3354
- adds available_device to test_cohen_kappa.py #3335 by @BanzaiTokyo in #3355
- adds available_device to test_confusion_matrix.py #3335 by @BanzaiTokyo in #3357
- Add SSIM for 3D tensors by @lrlunin in #3345
- adds available_device to test_entropy.py #3335 by @BanzaiTokyo in #3358
- adds available_device to test_kl_divergence.py #3335 by @BanzaiTokyo in #3363
- adds available_device to test_frequency.py #3335 by @BanzaiTokyo in #3359
- adds available_device to test_hsic.py #3335 by @BanzaiTokyo in #3361
- adds available_device to test_js_divergence.py #3335 by @BanzaiTokyo in #3362
- adds available_device to test_loss.py #3335 by @BanzaiTokyo in #3364
- adds available_device to test_maximum_mean_discrepancy #3335 by @BanzaiTokyo in #3365
- adds available_device to test_multilabel_confusion_matrix #3335 by @BanzaiTokyo in #3366
- Added support for logging dicts/iterables of metrics by @vfdev-5 in #3369
Acknowledgments
🎉 Thanks to our community and all our contributors for the issues, PRs and 🌟 ⭐️ 🌟 !
💯 We really appreciate your implication into the project
New Contributors
- @ZainRizvi made their first contribution in #3285
- @bonassifabio made their first contribution in #3309
- @Gulin7 made their first contribution in #3307
- @k22036 made their first contribution in #3320
- @jscanvic made their first contribution in #3332
- @HyeSungP made their first contribution in #3341
- @chaojessica made their first contribution in #3348
- @lrlunin made their first contribution in #3345
Full Changelog: v0.5.1...v0.5.2