From 2ae8f3aa522f20c2494541b71ecb846571fab3e3 Mon Sep 17 00:00:00 2001 From: vfdev Date: Wed, 4 Dec 2024 23:19:13 +0100 Subject: [PATCH] Update ignite/engine/__init__.py --- ignite/engine/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ignite/engine/__init__.py b/ignite/engine/__init__.py index 9c6c4a1bb71..6e82bc2f6bc 100644 --- a/ignite/engine/__init__.py +++ b/ignite/engine/__init__.py @@ -461,7 +461,7 @@ def create_supervised_trainer( amp_mode: can be ``amp`` or ``apex``, model and optimizer will be casted to float16 using `torch.cuda.amp `_ for ``amp`` and using `apex `_ for ``apex``. (default: None) - scaler: GradScaler instance for gradient scaling if `torch>=1.12.0` + scaler: GradScaler instance for gradient scaling if `torch>=1.6.0` and ``amp_mode`` is ``amp``. If ``amp_mode`` is ``apex``, this argument will be ignored. If True, will create default GradScaler. If GradScaler instance is passed, it will be used instead. (default: False)