updated config.py

IshanRattan · Aug 7, 2024 · 2bd6a6c · 2bd6a6c
1 parent ea8c7d6
commit 2bd6a6c
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/dl/image/classification/cifar10/config.py b/dl/image/classification/cifar10/config.py
@@ -3,4 +3,7 @@
 class_names = ('plane', 'car', 'bird', 'cat', 'deer', 'dog', 'frog', 'horse', 'ship', 'truck')
 data_path = './dataset/'
 batch_size = 64
-device = 'mps'
+device = 'mps'
+
+########################
+learning_rate = .01