Michael Diskin 4 ani în urmă
părinte
comite
9e97749f7d
1 a modificat fișierele cu 1 adăugiri și 1 ștergeri
  1. 1 1
      examples/albert/optim.py

+ 1 - 1
examples/albert/optim.py

@@ -108,7 +108,7 @@ class Lamb(Optimizer):
 
                 if self.adam:
                     trust_ratio = 1
-
+                logger.info("222")
                 p.data.add_(adam_step, alpha=-step_size * trust_ratio)
 
         return loss