Added loss functions.

2025-03-20 21:40:01 -07:00
parent 05196abbad
commit a6cefb1cf7
76 changed files with 1916 additions and 43 deletions
@@ -43,6 +43,6 @@ class PtApplyFunction:
            closure (Callable): Differentiable function to apply to the tensor.

        Returns:
-            Tuple[nn.Module]: A tuple containing the instantiated model.
+            Tuple[nn.Module]: A tuple containing the tensor.
        """
        return (closure(tens),)