We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent aa47c8c commit 40056c4Copy full SHA for 40056c4
tensor2tensor/data_generators/translate_enro.py
@@ -89,7 +89,7 @@ def dataset_splits(self):
89
"""Splits of data to produce and number of output shards for each."""
90
return [{
91
"split": problem.DatasetSplit.TRAIN,
92
- "shards": 1,
+ "shards": 16, # It's a small dataset, TPUs like at least a few shards.
93
}, {
94
"split": problem.DatasetSplit.EVAL,
95
"shards": 1,
0 commit comments