Update models/richard_model.py

models/richard_model.py

@@ -9,6 +9,6 @@ def get_richard_model():
     model.classifier[6] = nn.Linear(last_in, 512)
     model.classifier.add_module('new_relu', nn.ReLU())
     model.classifier.add_module('new_dropout', nn.Dropout())
-    model.classifier.add_module('new_linear', nn.Linear(512, num_classes))
+    model.classifier.add_module('new linear', nn.Linear(512, num_classes))
     
     return model