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

Commit 2ca3232

Browse files
Lukasz KaiserRyan Sepassi
Lukasz Kaiser
authored and
Ryan Sepassi
committed
Make default Parallelism in T2TModel reusing (removes "parallel_0" from variable names in colab).
PiperOrigin-RevId: 177265031
1 parent 8f05bab commit 2ca3232

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
@@ -75,7 +75,7 @@ def __init__(self,
7575
super(T2TModel, self).__init__(
7676
trainable=mode == tf.estimator.ModeKeys.TRAIN, name=name)
7777
if data_parallelism is None:
78-
data_parallelism = eu.Parallelism([""])
78+
data_parallelism = eu.Parallelism([""], reuse=True)
7979
if ps_devices is None:
8080
ps_devices = [""]
8181
if problem_hparams is None:

0 commit comments

Comments
 (0)