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)