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

Commit cc80721

Browse files
nshazeerRyan Sepassi
authored and
Ryan Sepassi
committed
Pass the modified hparams to the modalities, so they can know the mode.
PiperOrigin-RevId: 176688435
1 parent f77da80 commit cc80721

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tensor2tensor/utils/t2t_model.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -117,7 +117,7 @@ def __init__(self,
117117
self._ps_devices = ps_devices
118118
self._problem_hparams = problem_hparams
119119
self._problem_idx = problem_idx
120-
self._create_modalities(problem_hparams, hparams)
120+
self._create_modalities(problem_hparams, self._hparams)
121121

122122
def set_mode(self, mode):
123123
"""Set hparams with the given mode."""

0 commit comments

Comments
 (0)