Runtime Error when decoding about text-to-text-transfer-transformer HOT 5 CLOSED

can you try changing the flags to:

--gin_param="decode_from_file.input_filename='$DATA_DIR/testdata_pred.tsv'"
--gin_param="decode_from_file.output_filename='$DATA_DIR/testdata_outputs.txt'" 

from text-to-text-transfer-transformer.

It worked and produced an output. But at the end after the outputs were produced the script stalled at:

INFO:tensorflow:Waiting for new checkpoint at $MODEL_DIR

I can use it to produce what I want, but it's kinda odd that the script just stalls since I just told it to predict on that one checkpoint.

Also, FYI, the readme has another error in the instructions for invoking the decoder:
--gin_param == "utils.run.mode='infer'"

Should be changed to:
--gin_param="utils.run.mode='infer'"

This was a relatively easy error to find and fix though.

from text-to-text-transfer-transformer.

Thanks, I'll fix the README.

It actually should exit after outputting since you specified a checkpoint number. Can you paste the final command you used?

from text-to-text-transfer-transformer.

Can you try

--gin_param="utils.run.eval_checkpoint_step = 1005000"

and see if that fixes the stall?

from text-to-text-transfer-transformer.

Can you try

--gin_param="utils.run.eval_checkpoint_step = 1005000"

and see if that fixes the stall?

Yes, that worked, thanks! Looks like the argparse didn't put the arguments into the right places and it went with a default option?

from text-to-text-transfer-transformer.