1 files changed, 1 insertions, 1 deletions

diff --git a/training/functional.py b/training/functional.py
index 77f056e..ebb48ab 100644
--- a/training/functional.py
+++ b/training/functional.py
@@ -429,7 +429,7 @@ def train_loop(
     try:
         import dadaptation
 
-        isDadaptation = isinstance(optimizer, (dadaptation.DAdaptAdam, dadaptation.DAdaptAdan))
+        isDadaptation = isinstance(optimizer.optimizer, (dadaptation.DAdaptAdam, dadaptation.DAdaptAdan))
     except ImportError:
         pass