File size: 927 Bytes
75d4915 |
1 |
/home/ronakice/miniconda3/envs/py37/bin/t5_mesh_transformer --tpu=tpu-ronak-2 --gcp_project=justin-project-300310 --tpu_zone=europe-west4-a --model_dir=gs://pongo-bucket/duo-t5/experiments/38 --gin_file=gs://t5-data/pretrained_models/base/operative_config.gin --gin_file=infer.gin --gin_file=beam_search.gin --gin_param=utils.tpu_mesh_shape.tpu_topology = '2x2' --gin_param=infer_checkpoint_step = 1010900 --gin_param=utils.run.sequence_length = {'inputs': 1024, 'targets': 2} --gin_param=Bitransformer.decode.max_decode_length = 2 --gin_param=input_filename = 'gs://pongo-bucket/trec-covid-complete/data/6/duot5_input.txt' --gin_param=output_filename = 'gs://pongo-bucket/trec-covid-complete/data/6/duot5_output.txt' --gin_param=tokens_per_batch = 131072 --gin_param=Bitransformer.decode.beam_size = 1 --gin_param=Bitransformer.decode.temperature = 0.0 --gin_param=Unitransformer.sample_autoregressive.sampling_keep_top_k = -1 |