removed no_weight_decay from modeling_dass.py

modeling_dass.py

@@ -942,7 +942,7 @@ class SS2D(nn.Module):
             if merge:
                 A_log = A_log.flatten(0, 1)
         A_log = nn.Parameter(A_log)
-        A_log._no_weight_decay = True
+        #A_log._no_weight_decay = True
         return A_log
 
     @staticmethod
@@ -953,7 +953,7 @@ class SS2D(nn.Module):
             if merge:
                 D = D.flatten(0, 1)
         D = nn.Parameter(D)
-        D._no_weight_decay = True
+        #D._no_weight_decay = True
         return D
 
     @classmethod