learning/annotated_deep_learning_paper_implementations
1
0
Fork 0
mirror of https://github.com/labmlai/annotated_deep_learning_paper_implementations.git synced 2025-10-30 02:08:50 +08:00
Code Issues Packages Projects Releases Wiki Activity

hyper parameters

Browse Source
This commit is contained in:
Varuna Jayasiri
2022-08-26 16:13:31 +05:30
parent 0eb74d7b20
commit 1dbc2cbc04
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
labml_nn/normalization/deep_norm/experiment.py
View File

@ -74,7 +74,7 @@ class Configs(NLPAutoRegressionConfigs):
model: AutoregressiveTransformer
# Number of layers
n_layers: int = 64
n_layers: int = 32
# $\alpha$ and $\beta$ for DeepNorm
deep_norm_alpha: float
@ -153,9 +153,13 @@ def main():
# Switch between training and validation for $10$ times per epoch
'inner_iterations': 10,
# Number of layers
'n_layers': 50,
# Adam optimizer with no warmup
'optimizer.optimizer': 'Adam',
'optimizer.learning_rate': 3e-4,
'optimizer.learning_rate': 1.25e-4,
})
# Set model(s) for saving and loading