From d12244813e6a55d6434d7d021a1092c772c3405e Mon Sep 17 00:00:00 2001 From: sonyle4ka Date: Fri, 7 Mar 2025 20:54:55 +0300 Subject: [PATCH] =?UTF-8?q?=D0=A4=D0=BE=D0=BA=D0=B8=D0=BD=D0=B0=20=D0=A1?= =?UTF-8?q?=D0=BE=D1=84=D0=B8=D1=8F=20submission=5Ftemplate01.py?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- Deep Learning/submission_template01.py | 17 ++++++++++------- 1 file changed, 10 insertions(+), 7 deletions(-) diff --git a/Deep Learning/submission_template01.py b/Deep Learning/submission_template01.py index 1a8ded3..c1e4db6 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, bias=True), + nn.ReLU(), + nn.Linear(256, 16, bias=True), + nn.ReLU(), + nn.Linear(16, 10, bias=True) + ) + + return model def count_parameters(model): - # your code here - # return integer number (None is just a placeholder) - return None + return sum(p.numel() for p in model.parameters())