yujunjun/olmocr
1
0
Fork 0
mirror of https://github.com/allenai/olmocr.git synced 2025-10-12 16:52:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

More fixes

Browse Source
This commit is contained in:
Jake Poznanski 2025-05-29 17:20:32 +00:00
parent e8e6b6cb17
commit 15e0064212
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
scripts/run_benchmark.sh
View File

@ -45,7 +45,7 @@ image_tag = sys.argv[1]
beaker_user = sys.argv[2] beaker_user = sys.argv[2]
# Initialize Beaker client # Initialize Beaker client
b = Beaker.from_env(default_workspace="ai2/oe-data-pdf") b = Beaker.from_env(default_workspace="ai2/olmocr")
# Create experiment spec # Create experiment spec
experiment_spec = ExperimentSpec( experiment_spec = ExperimentSpec(
@ -69,14 +69,14 @@ experiment_spec = ExperimentSpec(
preemptible=True, preemptible=True,
), ),
resources=TaskResources(gpu_count=1), resources=TaskResources(gpu_count=1),
constraints=Constraint(cluster=["ai2/ceres-cirrascale", "ai2/jupiter-cirrascale"]), constraints=Constraints(cluster=["ai2/ceres-cirrascale", "ai2/jupiter-cirrascale"]),
result=ResultSpec(path="/noop-results"), result=ResultSpec(path="/noop-results"),
) )
], ],
) )
# Create the experiment # Create the experiment
experiment = b.experiment.create(spec=experiment_spec, workspace="ai2/oe-data-pdf") experiment = b.experiment.create(spec=experiment_spec, workspace="ai2/olmocr")
print(f"Created experiment: {experiment.id}") print(f"Created experiment: {experiment.id}")
print(f"View at: https://beaker.org/ex/{experiment.id}") print(f"View at: https://beaker.org/ex/{experiment.id}")
EOF EOF