Update deepmd/utils/finetune.py

Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com> Signed-off-by: Duo <[email protected]>
deepmodeling · Jun 11, 2024 · fd64ee5 · fd64ee5
1 parent c0d57e9
commit fd64ee5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepmd/utils/finetune.py b/deepmd/utils/finetune.py
@@ -38,7 +38,7 @@ def __init__(
         self.model_branch = model_branch
         self.random_fitting = random_fitting
         self.resuming = resuming
-        self.update_type = not (self.p_type_map == self.type_map)
+        self.update_type = self.p_type_map != self.type_map
 
     def get_index_mapping(self):
         """Returns the mapping index of newly defined types to those in the pretrained model."""