Updated MNISTModel to FastMNIST in workflow and check_model.py

shishir13 · Dec 5, 2024 · 390365a · 390365a
1 parent e8db2f6
commit 390365a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/MNIST_99.4/check_model.py b/MNIST_99.4/check_model.py
@@ -1,5 +1,5 @@
-from models.model import MNISTModel
+from models.model import FastMNIST
 from utils import print_model_summary
 
-model = MNISTModel()
+model = FastMNIST()
 print_model_summary(model)