set compress attr

Signed-off-by: Jinzhe Zeng <[email protected]>

Author: njzjz

deepmd/pt/model/descriptor/dpa2.py

@@ -310,6 +310,7 @@ def init_subclass_params(sub_data, sub_class):
         # set trainable
         for param in self.parameters():
             param.requires_grad = trainable
+        self.compress = False
 
     def get_rcut(self) -> float:
         """Returns the cut-off radius."""
@@ -888,7 +889,7 @@ def enable_compression(
             The overflow check frequency
         """
         # do some checks before the mocel compression process
-        if self.repinit.compress:
+        if self.compress:
             raise ValueError("Compression is already enabled.")
         assert (
             not self.repinit.resnet_dt