From f1e1595b3ae91a32011b2768460c8c641aa0bf95 Mon Sep 17 00:00:00 2001
From: yeziwen111 <303080700@qq.com>
Date: Fri, 28 Feb 2025 22:50:53 +0300
Subject: [PATCH] Update submission_template01.py

---
 Deep Learning/submission_template01.py | 19 +++++++++++--------
 1 file changed, 11 insertions(+), 8 deletions(-)

diff --git a/Deep Learning/submission_template01.py b/Deep Learning/submission_template01.py
index 1a8ded3..3b4d0b4 100644
--- a/Deep Learning/submission_template01.py	
+++ b/Deep Learning/submission_template01.py	
@@ -3,13 +3,16 @@
 from torch import nn
 
 def create_model():
-    # your code here
-    # return model instance (None is just a placeholder)
-
-    return None
+    model = nn.Sequential(
+            nn.Linear(784,256),
+            nn.ReLU(),
+            nn.Linear(256,16),
+            nn.ReLU(),
+            nn.Linear(16,10)
+    )    
+    return model
 
 def count_parameters(model):
-    # your code here
-    # return integer number (None is just a placeholder)
-    
-    return None
+    return sun(p.numel()for p in
+modle.parameters())               
+   # return integer number (None is just a placeholder)