Update eval.py

Kedreamix · Kedreamix · commit 82efb65e6d1c · 2022-05-08T17:28:46.000+08:00
diff --git a/CIFAR10_code/eval.py b/CIFAR10_code/eval.py
@@ -51,7 +51,7 @@ def eval_top5(outputs, label):
     elif args.net == 'ResNet34':
         from nets.ResNet import ResNet34
         net = ResNet34()
-    elif args.net == 'LeNet':
+    elif args.net == 'LeNet5':
         from nets.LeNet5 import LeNet5
         net = LeNet5()
     elif args.net == 'AlexNet':
