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())