pere commited on
Commit
91d785f
1 Parent(s): f226a89
__pycache__/tasks_v4.cpython-38.pyc CHANGED
Binary files a/__pycache__/tasks_v4.cpython-38.pyc and b/__pycache__/tasks_v4.cpython-38.pyc differ
 
finetune_large_mt5_sentencefix_v4.gin CHANGED
@@ -31,7 +31,7 @@ train_script.train:
31
  # `num_decodes` is equivalent to a beam size in a beam search decoding.
32
  models.EncoderDecoderModel.predict_batch_with_aux.num_decodes = 4
33
 
34
- partitioning.ModelBasedPjitPartitioner.num_partitions = 2
35
 
36
 
37
  #from t5.models import mesh_transformer
 
31
  # `num_decodes` is equivalent to a beam size in a beam search decoding.
32
  models.EncoderDecoderModel.predict_batch_with_aux.num_decodes = 4
33
 
34
+ partitioning.ModelBasedPjitPartitioner.num_partitions = 4
35
 
36
 
37
  #from t5.models import mesh_transformer