Skip to content

Commit

Permalink
fixed parameters
Browse files Browse the repository at this point in the history
  • Loading branch information
mg98 committed Nov 7, 2023
1 parent 0a0350a commit c429a8f
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions config.ini
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit c429a8f

Please sign in to comment.