From 00ac29b1d0e22de5ad5d5bb26ef2525651d95089 Mon Sep 17 00:00:00 2001 From: augustin64 Date: Fri, 17 Mar 2023 15:46:30 +0100 Subject: [PATCH] Update train.h parameters --- src/cnn/include/train.h | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/cnn/include/train.h b/src/cnn/include/train.h index 34e139e..5c19e36 100644 --- a/src/cnn/include/train.h +++ b/src/cnn/include/train.h @@ -5,9 +5,9 @@ #define DEF_TRAIN_H #define EPOCHS 10 -#define BATCHES 500 +#define BATCHES 32 #define USE_MULTITHREADING -#define LEARNING_RATE 0.05 +#define LEARNING_RATE 3e-4 /*