t5x.main binary

Contents

t5x.main binary#

The main entrance for running any of the T5X supported binaries.

Currently this includes train/infer/eval/precompile.

Example Local (CPU) Pretrain Gin usage

python -m t5x.main: –gin_file=t5x/examples/t5/t5_1_1/tiny.gin –gin_file=t5x/configs/runs/pretrain.gin –gin.MODEL_DIR="/tmp/t5x_pretrain" –gin.TRAIN_STEPS=10 –gin.MIXTURE_OR_TASK_NAME="c4_v220_span_corruption" –gin.MIXTURE_OR_TASK_MODULE="t5.data.mixtures" –gin.TASK_FEATURE_LENGTHS=”{‘inputs’: 128, ‘targets’: 30}” –gin.DROPOUT_RATE=0.1 –run_mode=train –logtostderr

class t5x.main.RunMode(value)[source]#: All the running mode possible in T5X.