Skip to content

Commit

Permalink
Correct eval_checkpoint_step param flag in README. Fixes #5
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 277614751
  • Loading branch information
adarob authored and t5-copybara committed Oct 30, 2019
1 parent ca22fd3 commit 7f62dfb
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -158,13 +158,13 @@ t5_mesh_transformer \
--gin_file="beam_search.gin" \
--gin_param="utils.tpu_mesh_shape.tpu_topology = '2x2'" \
--gin_param="MIXTURE_NAME = 'glue_mrpc_v002'" \
--gin_param="eval_checkpoint_step = 'all'"
--gin_param="utils.run.eval_checkpoint_step = 'all'"
```

To evaluate a specific checkpoint, simply set the `eval_checkpoint_step` parameter to appropriate checkpoint.

```
--gin_param="eval_checkpoint_step = 100000"
--gin_param="utils.run.eval_checkpoint_step = 100000"
```

You can also use `greedy_decode.gin` or `sample_decode.gin` instead of `beam_search.gin` in the command above.
Expand All @@ -186,13 +186,13 @@ t5_mesh_transformer \
--gin_param="decode_from_file.input_filename = '/path/to/inputs.txt'"\
--gin_param="decode_from_file.output_filename = '/tmp/outputs.txt'"\
--gin_param="utils.tpu_mesh_shape.tpu_topology = '2x2'"\
--gin_param="eval_checkpoint_step = 'all'"
--gin_param="utils.run.eval_checkpoint_step = 'all'"
```

To predict with a specific checkpoint, simply set the `eval_checkpoint_step` parameter to appropriate checkpoint.

```
--gin_param="eval_checkpoint_step = 100000"
--gin_param="utils.run.eval_checkpoint_step = 100000"
```

You can also use `beam_search.gin` or `greedy_decode.gin` instead of `sample_decode.gin` in the command above.
Expand Down

0 comments on commit 7f62dfb

Please sign in to comment.