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