diff --git a/benchmarks/torchvision/voirfile.py b/benchmarks/torchvision/voirfile.py index fe79714fb..72bc1b9ca 100644 --- a/benchmarks/torchvision/voirfile.py +++ b/benchmarks/torchvision/voirfile.py @@ -54,7 +54,7 @@ def instrument_main(ov, options: Config): batch_size_fn=lambda x: len(x[0]) ) - probe = ov.probe(f"//{refstring(imagenet_dataloader)}() as loader", overridable=True) + probe = ov.probe(f"{refstring(imagenet_dataloader)}() as loader", overridable=True) probe['loader'].override(wrapper.loader) probe = ov.probe("//train_epoch > criterion", overridable=True) diff --git a/scripts/article/run_cuda.sh b/scripts/article/run_cuda.sh index 048e7dd58..3ac8cb36a 100644 --- a/scripts/article/run_cuda.sh +++ b/scripts/article/run_cuda.sh @@ -62,6 +62,9 @@ fi cd $MILABENCH_WORDIR +(cd $MILABENCH_WORDIR/milabench && git pull origin intel) +pip install -e $MILABENCH_WORDIR/milabench + # # Run the benchmakrs milabench run "$@"