exp

finetune_large_mt5_sentencefix_v4_16.gin

@@ -31,10 +31,10 @@ utils.RestoreCheckpointConfig:
   path = %INITIAL_CHECKPOINT_PATH
   mode = 'specific'
 
-train_script.train:
-  train_dataset_cfg = @train/utils.DatasetConfig()
-  train_eval_dataset_cfg = @train_eval/utils.DatasetConfig()
-  infer_eval_dataset_cfg = @infer_eval/utils.DatasetConfig()
+#train_script.train:
+#  train_dataset_cfg = @train/utils.DatasetConfig()
+#  train_eval_dataset_cfg = @train_eval/utils.DatasetConfig()
+#  infer_eval_dataset_cfg = @infer_eval/utils.DatasetConfig()
 
 models.EncoderDecoderModel.predict_batch_with_aux.num_decodes = 4
 
@@ -50,5 +50,5 @@ infer_eval/utils.DatasetConfig:
   module = %MIXTURE_OR_TASK_MODULE
 
 
-partitioning.PjitPartitioner.num_partitions = 4
+#partitioning.PjitPartitioner.num_partitions = 4