Skip to content
This repository was archived by the owner on Jul 7, 2023. It is now read-only.

Commit 1cd3f25

Browse files
author
Ryan Sepassi
committed
Revert hparam change to versioned set
PiperOrigin-RevId: 184216942
1 parent 07d8308 commit 1cd3f25

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

tensor2tensor/models/transformer.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -834,7 +834,7 @@ def transformer_base_v1():
834834
hparams.initializer = "uniform_unit_scaling"
835835
hparams.weight_decay = 0.0
836836
hparams.optimizer_adam_beta1 = 0.9
837-
hparams.optimizer_adam_beta2 = 0.997
837+
hparams.optimizer_adam_beta2 = 0.98
838838
hparams.num_sampled_classes = 0
839839
hparams.label_smoothing = 0.1
840840
hparams.shared_embedding_and_softmax_weights = True
@@ -883,7 +883,11 @@ def transformer_base_v2():
883883

884884
@registry.register_hparams
885885
def transformer_base():
886-
return transformer_base_v2()
886+
# Update parameters here, then occasionally cut a versioned set, e.g.
887+
# transformer_base_v2.
888+
hparams = transformer_base_v2()
889+
hparams.optimizer_adam_beta2 = 0.997
890+
return hparams
887891

888892

889893
@registry.register_hparams

0 commit comments

Comments
 (0)