test
Browse files
finetune_large_mt5_sentencefix_v4_16.gin
CHANGED
|
@@ -33,6 +33,7 @@ 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
|
| 38 |
#import t5.models
|
|
|
|
| 33 |
|
| 34 |
#partitioning.ModelBasedPjitPartitioner.num_partitions = 2
|
| 35 |
|
| 36 |
+
partitioning.PjitPartitioner.num_partitions = 2
|
| 37 |
|
| 38 |
#from t5.models import mesh_transformer
|
| 39 |
#import t5.models
|