Skip to content

Commit

Permalink
Update config_transformer.py
Browse files Browse the repository at this point in the history
  • Loading branch information
lvapeab committed May 26, 2020
1 parent 81c72c0 commit 3067e59
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion examples/configs/config_transformer.py
Original file line number Diff line number Diff line change
Expand Up @@ -138,7 +138,7 @@ def load_parameters():
LR_GAMMA = 1 # Multiplier used for decreasing the LR
LR_REDUCE_EACH_EPOCHS = False # Reduce each LR_DECAY number of epochs or updates
LR_START_REDUCTION_ON_EPOCH = 0 # Epoch to start the reduction
LR_REDUCER_TYPE = 'Noam' # Function to reduce. 'linear' and 'exponential' implemented.
LR_REDUCER_TYPE = 'noam' # Function to reduce. 'linear' and 'exponential' implemented.
# Linear reduction: new_lr = lr * LR_GAMMA
# Exponential reduction: new_lr = lr * LR_REDUCER_EXP_BASE ** (current_nb / LR_HALF_LIFE) * LR_GAMMA
# Noam reduction: new_lr = lr * min(current_nb ** LR_REDUCER_EXP_BASE, current_nb * LR_HALF_LIFE ** WARMUP_EXP)
Expand Down

0 comments on commit 3067e59

Please sign in to comment.