pere
/

eu-jav-categorisation

pere commited on Apr 1, 2022

Commit

8dc8092

•

1 Parent(s): 5fad4d8

test

Files changed (3) hide show

__pycache__/tasks.cpython-38.pyc CHANGED Viewed

Binary files a/__pycache__/tasks.cpython-38.pyc and b/__pycache__/tasks.cpython-38.pyc differ

finetuning_categorisation.gin CHANGED Viewed

@@ -29,7 +29,7 @@ INITIAL_CHECKPOINT_PATH = "gs://t5-data/pretrained_models/t5x/mt5_base/checkpoin
 #train_script.train:
 #  eval_period = 500
 #  partitioner = @partitioning.ModelBasedPjitPartitioner()
-# partitioning.ModelBasedPjitPartitioner.num_partitions = 2
 # `num_decodes` is equivalent to a beam size in a beam search decoding.
 models.EncoderDecoderModel.predict_batch_with_aux.num_decodes = 4

 #train_script.train:
 #  eval_period = 500
 #  partitioner = @partitioning.ModelBasedPjitPartitioner()
+# partitioning.PjitPartitioner.num_partitions = 1
 # `num_decodes` is equivalent to a beam size in a beam search decoding.
 models.EncoderDecoderModel.predict_batch_with_aux.num_decodes = 4

finetuning_categorisation_xxl.gin CHANGED Viewed

@@ -30,7 +30,7 @@ INITIAL_CHECKPOINT_PATH = "gs://t5-data/pretrained_models/t5x/mt5_xxl/checkpoint
 #train_script.train:
 #  eval_period = 500
 #  partitioner = @partitioning.ModelBasedPjitPartitioner()
-partitioning.ModelBasedPjitPartitioner.num_partitions = 1
 # `num_decodes` is equivalent to a beam size in a beam search decoding.
 models.EncoderDecoderModel.predict_batch_with_aux.num_decodes = 4

 #train_script.train:
 #  eval_period = 500
 #  partitioner = @partitioning.ModelBasedPjitPartitioner()
+partitioning.PjitPartitioner.num_partitions = 1
 # `num_decodes` is equivalent to a beam size in a beam search decoding.
 models.EncoderDecoderModel.predict_batch_with_aux.num_decodes = 4