ignite
ignite copied to clipboard
High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.
Hi! I'm one of the maintainers of [Pyrefly](https://github.com/facebook/pyrefly), which is a Python type checker and language server. We've recently migrated a couple projects in the pytorch ecosystem (https://github.com/pytorch/helion/pull/1143, https://github.com/pytorch/pytorch/pull/166197) to...
``` ____________________________ test_auto_methods_hvd _____________________________ gloo_hvd_executor = @pytest.mark.distributed @pytest.mark.skipif(not idist.has_hvd_support, reason="Skip if no Horovod dist support") @pytest.mark.skipif("WORLD_SIZE" in os.environ, reason="Skip if launched as multiproc") def test_auto_methods_hvd(gloo_hvd_executor): device = "cpu" if not...
Fixes #1521 Depends on: - https://github.com/pytorch/ignite/pull/3442 - https://github.com/pytorch/ignite/pull/3443 - https://github.com/pytorch/ignite/pull/3444 - https://github.com/pytorch/ignite/pull/3445