ignite
ignite copied to clipboard
pytorch
High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.
Let's try to use https://docs.astral.sh/uv instead of conda in this CI job: - [ ] https://github.com/pytorch/ignite/blob/41d75522eb36d4b9c55afea941ec53f42430315e/.github/workflows/pytorch-version-tests.yml - [ ] https://github.com/pytorch/ignite/blob/41d75522eb36d4b9c55afea941ec53f42430315e/.github/workflows/unit-tests.yml We can use GHA: actions/setup-uv - **Each task should be...
from ignite.metrics import Metric ImportError: cannot import name 'Metric' from partially initialized module 'ignite.metrics' (most likely due to a circular import) /ignite/metrics/__init__.py) I solved that by def create_supervised_evaluator( model: torch.nn.Module,...
Adds NDCG metrics as a part of adding RecSys metrics in #2631 This PR builds on the work done in #3054 and #2632 (much thanks to ili0820 and kamalojasv181!) The...
=> stable pytorch GPU tests are still failing on `tests/ignite/metrics/test_classification_report.py::test_distrib_nccl_gpu Fatal Python error: Aborted`
We've flipped the default for the weights_only argument in torch.load to True in pytorch/pytorch, see [here](https://dev-discuss.pytorch.org/t/bc-breaking-change-torch-load-is-being-flipped-to-use-weights-only-true-by-default-in-the-nightlies-after-137602/2573) for details + [documentation](https://pytorch.org/docs/main/notes/serialization.html#torch-load-with-weights-only-true) and this is coming in torch 2.6. This is expected...
