shellyriver
/

test1

shellyriver commited on Jul 18, 2023

Commit

dac6f0a

1 Parent(s): 0650a36

Update model.py

Files changed (1) hide show

model.py CHANGED Viewed

@@ -36,6 +36,6 @@ class CNN(nn.Module):
         return x
 def get_model():
-    return CNN

         return x
 def get_model():
+    return CNN(num_channel=1, num_classes=10, num_pixel=28)