From 7934ba6abab6ddfffc4f503158003f04e67b235d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Stardust=C2=B7=E5=87=8F?= <2225664821@qq.com>
Date: Fri, 25 Aug 2023 20:59:46 +0800
Subject: [PATCH] Update utils.py

---
 utils.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/utils.py b/utils.py
index fab8ce7..04a05cb 100644
--- a/utils.py
+++ b/utils.py
@@ -22,8 +22,8 @@ def load_checkpoint(checkpoint_path, model, optimizer=None, skip_optimizer=False
     learning_rate = checkpoint_dict['learning_rate']
     if optimizer is not None and not skip_optimizer and checkpoint_dict['optimizer'] is not None:
         optimizer.load_state_dict(checkpoint_dict['optimizer'])
-    elif optimizer is None and not skip_optimizer:  #Disable this line if Infer ,and enable the line below
-    #else:
+    elif optimizer is None and not skip_optimizer:  
+    #else: #Disable this line if Infer ,and enable the line upper
         new_opt_dict = optimizer.state_dict()
         new_opt_dict_params = new_opt_dict['param_groups'][0]['params']
         new_opt_dict['param_groups'] = checkpoint_dict['optimizer']['param_groups']