update docstring

lululxvi · Dec 9, 2024 · c773a2f · c773a2f
1 parent c7e909a
commit c773a2f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/deepxde/nn/regularizers.py b/deepxde/nn/regularizers.py
@@ -2,7 +2,7 @@
 
 
 def get(identifier):
-    """Retrieves a TensorFlow regularizer instance based on the given identifier.
+    """Retrieves a regularizer instance based on the given identifier.
 
     Args:
         identifier (list/tuple): Specifies the type and factor(s) of the regularizer.
@@ -27,7 +27,7 @@ def get(identifier):
     if name == "l2":
         return bkd.l2_regularization(factor[0])
     if name in ("l1l2", "l1+l2"):
-        # TODO: only supported by 'tensorflow.compat.v1' now.
+        # TODO: only supported by 'tensorflow.compat.v1' and 'tensorflow' now.
         if len(factor) < 2:
             raise ValueError("L1L2 regularizer requires both L1/L2 penalties.")
         return bkd.l1_l2_regularization(factor[0], factor[1])