DylanJHJ's picture
add t5 mesh tensorflow checkpoint
36cb93b
raw
history blame contribute delete
985 Bytes
/home/cfdaclip/.local/bin/t5_mesh_transformer --tpu=node-1 --gcp_project=convrerank --tpu_zone=us-central1-f --model_dir=gs://convrerank/checkpoints/monot5-base-canard4ir-t40 --gin_file=gs://t5-data/pretrained_models/base/operative_config.gin --gin_file=infer.gin --gin_file=score_from_file.gin --gin_param=utils.tpu_mesh_shape.tpu_topology = '2x2' --gin_param=infer_checkpoint_step = 1140400 --gin_param=utils.run.sequence_length = {'inputs': 512, 'targets': 4} --gin_param=Bitransformer.decode.max_decode_length = 2 --gin_param=inputs_filename = 'gs://convrerank/rerank_cast2019/input/cast2019.train.cqe.conv.rerank.txt' --gin_param=targets_filename = 'gs://convrerank/true.txt' --gin_param=scores_filename = 'gs://convrerank/rerank_cast2019/output/ablation_train/cast2019.train.cqe.conv.rerank.t40-1140400.txt' --gin_param=Bitransformer.decode.beam_size = 1 --gin_param=Bitransformer.decode.temperature = 0.0 --gin_param=Unitransformer.sample_autoregressive.sampling_keep_top_k = -1