another partitioner test
Browse files
finetune_large_mt5_sentencefix_v4_16.gin
CHANGED
|
@@ -26,7 +26,6 @@ INITIAL_CHECKPOINT_PATH = "gs://t5-data/pretrained_models/t5x/mt5_large/checkpoi
|
|
| 26 |
|
| 27 |
train_script.train:
|
| 28 |
eval_period = 500
|
| 29 |
-
partitioner = @partitioning.ModelBasedPjitPartitioner()
|
| 30 |
|
| 31 |
# `num_decodes` is equivalent to a beam size in a beam search decoding.
|
| 32 |
models.EncoderDecoderModel.predict_batch_with_aux.num_decodes = 4
|
|
|
|
| 26 |
|
| 27 |
train_script.train:
|
| 28 |
eval_period = 500
|
|
|
|
| 29 |
|
| 30 |
# `num_decodes` is equivalent to a beam size in a beam search decoding.
|
| 31 |
models.EncoderDecoderModel.predict_batch_with_aux.num_decodes = 4
|