update

Author: cocoocoder

mftcoder_accelerate/src/utils/loss_utils.py

@@ -310,7 +310,7 @@ def compute_per_task_weight(self, completed_steps=None):
 
         return per_task_weight
 
-    def fit_window_slope(self, x, y, type='slope'):
+    def fit_window_slope(self, x, y, type="slope"):
 
         y = y[y != 0]
         x = x[:len(y)]
@@ -323,7 +323,7 @@ def fit_window_slope(self, x, y, type='slope'):
         ws = ws.double()
 
         if len(y) >= 2:
-            if type == 'slope':
+            if type == "slope":
                 X = torch.stack((x, torch.ones_like(x, dtype=torch.float64))).T
                 X = X.double()
             else: