YogeLiu/TexTeller
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update README

Browse Source
This commit is contained in:
三洋三洋
2024-05-07 07:28:16 +00:00
parent ff1872d067
commit ec90b2fdb9
2 changed files with 12 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
README.md
View File

@@ -160,11 +160,13 @@ If you are using a different dataset, you might need to retrain the tokenizer to
### Training the Model
To train the model, you need to run the following command in the `TexTeller/src` directory:
1. Modify `num_processes` in `src/train_config.yaml` to match the number of GPUs available for training (default is 1).
```bash
python -m models.ocr_model.train.train
```
2. In the `TexTeller/src` directory, run the following command:
```bash
accelerate launch --config_file ./train_config.yaml -m models.ocr_model.train.train
```
You can set your own tokenizer and checkpoint paths in `TexTeller/src/models/ocr_model/train/train.py` (refer to `train.py` for more information). If you are using the same architecture and dictionary as TexTeller, you can also fine-tune TexTeller's default weights with your own dataset.