update on linear

pouyam79i · Mar 30, 2024 · 6c5fb01 · 6c5fb01
1 parent f669c69
commit 6c5fb01
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Project 1/Phase 2/mytorch-framework/mytorch/layer/linear.py b/Project 1/Phase 2/mytorch-framework/mytorch/layer/linear.py
@@ -30,4 +30,4 @@ def parameters(self):
         return ...
 
     def __str__(self) -> str:
-        return "linear"
+        return "linear - total param: {} - in: {}, out: {}".format(self.inputs * self.outputs, self.inputs, self. outputs)