diff --git a/dl/image/classification/cifar10/config.py b/dl/image/classification/cifar10/config.py index 0087671..7f495fd 100644 --- 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' \ No newline at end of file +device = 'mps' + +######################## +learning_rate = .01 \ No newline at end of file