Fix for https://github.com/NVIDIA/apex/issues/344

04667139 · Michael Carilli · 1dca16cc · 04667139
Commit 04667139 authored Jun 07, 2019 by Michael Carilli
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletion

apex/amp/frontend.py apex/amp/frontend.py +4 -1

No files found.
--- a/apex/amp/frontend.py
+++ b/apex/amp/frontend.py
@@ -308,7 +308,10 @@ def initialize(
    _amp_state.verbosity = verbosity

    if not enabled:
-        return models, optimizers
+        if optimizers is None:
+            return models
+        else:
+            return models, optimizers

    if not torch.backends.cudnn.enabled:
        raise RuntimeError(