diff --git a/config.ini b/config.ini index 362c52e..a49b6ce 100644 --- a/config.ini +++ b/config.ini @@ -15,14 +15,14 @@ Optimizer = SGD EpochScale = 1 # Learning rate for the optimizer -LearningRate = 0.0006 +LearningRate = 0.0007 # Number of hidden layers and units per layer HiddenLayers = 2 HiddenUnits = 1024 # Regularization through dropout to prevent overfitting (set 0 to disable) -Dropout = 0.2 +Dropout = 0.3 # Enable quantization-aware training and quantized model exchange Quantize = False