| notepads | ||
| old | ||
| utils | ||
| .gitignore | ||
| config-eval.yaml | ||
| config-train.yaml | ||
| eval.py | ||
| README.md | ||
| requirements.txt | ||
| single_eval.py | ||
| test.image.jpg | ||
| train.py | ||
Donut script for train and eval
Instalation/setup:
-> pip install -r requirements.txt
-> export HUG_TOKEN=YourToken - token needed for pushing model to huggingface after training
Usage:
Train
-> python train --config config-train.yaml
Evaluation
-> python eval --config config-eval.yaml
Configuration
Need to suply config file like config-eval.yaml for evaluation or config-train.yaml for traning