github pytorch/ignite v0.4.5
New metrics, extended DDP support and bug fixes

latest releases: v0.5.0.post2, v0.5.0.post1, v0.4.13...
2 years ago

PyTorch-Ignite 0.4.5 - Release Notes

New Features

Metrics

  • Added BLEU metric (#1834)
  • Added ROUGE metric (#1772)
  • Added MultiLabelConfusionMatrix metric (#1613)
  • Added Cohen Kappa metric (#1690)
  • Extended sync_all_reduce API (#1823)
  • Made EpochMetric more generic by extending the list of valid types (#1748)
  • Fixed issue with metric's output device (#2062)
  • Added support for list of tensors as metric input (#2055)
  • Implemented Jaccard Index shortcut for metrics (#1682)
  • Updated Loss metric to use required_output_keys (#2027)
  • Added classification report metric (#1887)
  • Added output detach for Canberra metric (#1820)
  • Improved ROC AUC (#1762)
  • Improved AveragePrecision metric and tests (#1756)
  • Uniformly handling of metric types for all loggers (#2021)
  • More DDP support for multiple contrib metrics (#1891, #1869, #1865, #1850, #1830, #1829, #1806, #1805, #1803)

Engine

  • Added native torch.cuda.amp and apex automatic mixed precision for create_supervised_trainer and create_supervised_evaluator (#1714, #1589)
  • Updated state.batch/state.output lifespan in Engine (#1919)

Distributed module

  • Handled IterableDataset with auto_dataloader (#2028)
  • Updated Loss metric to use required_output_keys (#2027)
  • Enabled gpu support for gloo backend (#2016)
  • Added safe_mode for idist broadcast (#1839)
  • Improved idist to support different init_methods (#1767)

Other improvements

  • Added LR finder improvements, moved to core (#2045, #1998, #1996, #1987, #1981, #1961, #1951, #1930)
  • Moved param handler to core (#1988)
  • Added an option to store EpochOutputStore data on engine.state, moved to core (#1982, #1974)
  • Set seed for xla in ignite.utils.manual_seed (#1970)
  • Fixed case for Precision/Recall in multi_label, not averaged configuration for DDP (#1646)
  • Updated PolyaxonLogger to handle v1 and v0 (#1625)
  • Added Arguments *args, **kwargs to BaseLogger.attach method (#2034)
  • Enabled metric ordering on ProgressBar (#1937)
  • Updated wandb logger (#1896)
  • Fixed type hint for ProgressBar (#2079)

Bug fixes

  • BC-breaking: Improved loggers to keep configuration (#1945)
  • Fixed warnings in CI (#2023)
  • Fixed Precision for all zero predictions (#2017)
  • Renamed the default logger (#2006)
  • Fixed Accumulation metric with Nvidia/Apex (#1978)
  • Updated code to raise an error if SLURM is used with torch dist launcher (#1976)
  • Updated nltk-smooth2 for BLEU metric (#1911)
  • Added full read permissions to saved file (1876) (#1880)
  • Fixed a bug with horovod _do_manual_all_reduce (#1848)
  • Fixed small bug in "Finetuning EfficientNet-B0 on CIFAR100" tutorial (#2073)
  • Fixed f-string in mnist_save_resume_engine.py example (#2077)
  • Fixed an issue when rng states accidentaly on cuda for DeterministicEngine (#2081)

Housekeeping

A lot of PRs

Acknowledgments

🎉 Thanks to our community and all our contributors for the issues, PRs and 🌟 ⭐️ 🌟 !
💯 We really appreciate your implication into the project (in alphabetical order):

@01-vyom, @Devanshu24, @Juddd, @KickItLikeShika, @Moh-Yakoub, @Muktan, @OBITORASU, @Priyansi, @afzal442, @ahmedo42, @aksg87, @aniezurawski, @cozek, @devrimcavusoglu, @fco-dv, @gucifer, @log-layer, @mouradmourafiq, @radekosmulski, @sahilg06, @sdesrozis, @sparkingdark, @thomasjpfan, @touqir14, @trsvchn, @vfdev-5, @ydcjeff

Don't miss a new ignite release

NewReleases is sending notifications on new releases.