yujunjun/olmocr
1
0
Fork 0
mirror of https://github.com/allenai/olmocr.git synced 2025-06-27 04:00:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

weka

Browse Source
This commit is contained in:
Jake Poznanski 2025-06-27 02:57:26 +00:00
parent 680377c93f
commit 1451dd1395
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
scripts/train/newtrainer-gantry.sh
View File

@ -60,6 +60,8 @@ gantry run \
--env-secret AWS_ACCESS_KEY_ID=S2_AWS_ACCESS_KEY_ID \
--env-secret AWS_SECRET_ACCESS_KEY=S2_AWS_SECRET_ACCESS_KEY \
--env-secret WANDB_API_KEY=JAKE_WANDB_API_KEY \
--weka oe-data-default:/weka/oe-data-default \
--weka oe-training-default:/weka/oe-training-default \
--shared-memory 10GiB \
--yes \
-- /bin/bash -c "source scripts/beaker/jupiter-ib.sh && python -m olmocr.train.train --config olmocr/train/configs/example_config.yaml"