Skip to content

Commit

Permalink
Update README
Browse files Browse the repository at this point in the history
  • Loading branch information
avaucher committed Oct 5, 2023
1 parent e8f9bca commit 40da86d
Showing 1 changed file with 10 additions and 0 deletions.
10 changes: 10 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -123,3 +123,13 @@ rxn-onmt-finetune --model_output_dir $MODELS_FINETUNED --preprocess_dir $PREPROC
```
The syntax is very similar to `rxn-onmt-train` and `rxn-onmt-continue-training`.
This is compatible both with single-task and multi-task.


### Saving space by stripping the models

The model files can be quite large (~250 MB) for the default architecture.
Roughly two thirds of the size is taken by the optimizer state, which is usually not needed, except maybe for the last checkpoint.
You can strip the models with the command
```shell
rxn-onmt-strip-checkpoints -m $MODELS
```

0 comments on commit 40da86d

Please sign in to comment.