ignite
ignite copied to clipboard
pytorch
High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.
### Feature Automatic mixed precision for xla has landed in [pytorch 1.8.1](https://github.com/pytorch/pytorch/releases/tag/v1.8.1) and torch/xla nightly. We should enable it in `create_supervised_*` helper functions. ### Suggested solution Remove xla and amp...
Our project code coverage start to decrease with new features and reworked tests. We are now ~95% but were 96% recently and around ~98% the last year... The idea is...
## 🚀 Feature Following https://github.com/pytorch/ignite/pull/1700#issuecomment-787151203 we would like to improve `EpochMetric` by supporting more output types returned by `compute_fn`: - EpochMetric should work with compute_fn outputs: scalars, tensors, tuple/list/mapping of...
## 🚀 Feature The idea is to add "oneCCL" backend by extending our [torch native dist model](https://github.com/pytorch/ignite/blob/395743d1f782ca84a7fb1a3c71afe2fa76b882a1/ignite/distributed/comp_models/native.py#L22). Any interest for that ? Refs: - https://pytorch.medium.com/optimizing-dlrm-by-using-pytorch-with-oneccl-backend-9f85b8ef6929 - https://github.com/intel/torch-ccl
Currently, the testing data is generated without taking into account distributed context: https://github.com/pytorch/ignite/blob/e17acc740d38909ddbbed6648423750abe593e58/tests/ignite/metrics/test_confusion_matrix.py#L559-L565 same for MultiLabelConfusionMatrix tests. We would like to rewrite the tests in a similar way as here...
## 🚀 Feature Currently, we can build and publish dockers using Circle CI and it works more or less well, however, PR's modifying Dockerfile can not build docker images to...
## 🚀 Feature The idea is to verify the output type for all metrics (output of `compute` function) and update the docs accordingly. In general, metric's output should be a...
In full detail the feature request is described [here](https://github.com/pytorch/pytorch/issues/22439#issuecomment-636454497), below is a quick recap. There are two inconveniences I experience with the current interface of Accuracy. #### 1. Inconsistent input...
## 🚀 Feature - Livelossplot logger The logger that is similar to Neptune, W&B, tensorboard loggers available in Ignite. livelossplot: [https://github.com/stared/livelossplot](url) ignite plugin that is available in the livelossplot:[https://github.com/stared/livelossplot/blob/master/livelossplot/inputs/pytorch_ignite.py](url) notebook...
## 🚀 Feature Idea is to provide Expected Calibration Error metrics as coded in Baal : https://github.com/ElementAI/baal/blob/master/src/baal/utils/metrics.py#L88 cc @Dref360