From c11f30e1c1a5f27504011b746c67cdc9853c8d12 Mon Sep 17 00:00:00 2001 From: Volpeon Date: Wed, 22 Mar 2023 08:30:55 +0100 Subject: Fix --- training/functional.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'training') 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 -- cgit v1.2.3-70-g09d2