ignite
ignite copied to clipboard
pytorch
High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.
## 🚀 Feature There can be use-cases when we would like to get trainer's epoch/iteration or/and other items from `trainer.state`. Let's propose an API such that we could get easily...
Following ["Layer rotation: a surprisingly powerful indicator of generalization in deep networks?" paper](https://arxiv.org/pdf/1806.01603v2.pdf), layer rotation indicator provides quantative measure on generalization. We can integrate a LayerRotation indicator handler based on...
## 🚀 Feature TorchSnapshot is a performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind. It includes many optimizations to control for memory usage...
Description: Update horovod to take all_reduce with group Check list: - [ ] New tests are added (if a new feature is added) - [ ] New doc strings: description...
Description: added maximum mean discrepancy (MMD) metric, which measures the discrepancy of two distributions. https://www.onurtunali.com/ml/2019/03/08/maximum-mean-discrepancy-in-machine-learning.html Check list: - [x] New tests are added (if a new feature is added) -...
@vfdev-5 This is an attempt to rerun some of the CI tests using the retry github action from nick-fields. The objective is to rerun steps in the GitHub workflows that...
By default, the encoding is `utf-8` for compatibility with CJK characters. Fixes #3239 Description: Add an optional arg `encoding` to `setup_logger()`. By default, the `encoding` is `utf-8` for compatibility with...
Supercedes #1565, I failed to merge/rebase due to conflicts. Description: - bug fixes - state_dict/load_state_dict refactor - added helper Engine.debug() method - more tests Check list: - [ ] New...
## 🚀 Feature Add optional arg `encoding = None` for `ignite.utils.setup_logger` ## Motivation I encountered garbled text when loggering messages that contains CJK characters using `logger.info()`. For example, ```python from...