stop checkpoints polluting git history ;)

.gitignore

+# Pytorch Checkpoints
+*.pth 
\ No newline at end of file

mlp_train.py

@@ -208,6 +208,9 @@ def main():
     X, y = prepare_data()
     model = train_model(X, y)
 
+    # print parameter count of model
+    print(f"Parameter count: {sum(p.numel() for p in model.parameters())}")
+
 
 if __name__ == "__main__":
     main()