|
from __gin__ import dynamic_registration |
|
import tasks |
|
|
|
import __main__ as infer_script |
|
from t5.data import mixtures |
|
from t5x import partitioning |
|
from t5x import utils |
|
|
|
include "t5x/examples/t5/mt5/base.gin" |
|
include "t5x/configs/runs/infer.gin" |
|
|
|
DROPOUT_RATE = 0.0 |
|
MIXTURE_OR_TASK_NAME = "sentencefix" |
|
TASK_FEATURE_LENGTHS = {"inputs": 256, "targets": 256} |
|
|
|
infer_script.infer: |
|
partitioner = @partitioning.ModelBasedPjitPartitioner() |
|
|
|
partitioning.ModelBasedPjitPartitioner.num_partitions = 1 |
|
|
|
utils.DatasetConfig: |
|
split = "test" |
|
batch_size = 32 |
|
|
