Skip to content

Commit 40056c4

Browse files
Lukasz Kaiserkpe
Lukasz Kaiser
authored andcommitted
Have a few shards for en-ro translation.
PiperOrigin-RevId: 224177849
1 parent aa47c8c commit 40056c4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tensor2tensor/data_generators/translate_enro.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -89,7 +89,7 @@ def dataset_splits(self):
8989
"""Splits of data to produce and number of output shards for each."""
9090
return [{
9191
"split": problem.DatasetSplit.TRAIN,
92-
"shards": 1,
92+
"shards": 16, # It's a small dataset, TPUs like at least a few shards.
9393
}, {
9494
"split": problem.DatasetSplit.EVAL,
9595
"shards": 1,

0 commit comments

Comments
 (0)