-

mila-iqia · Jun 6, 2024 · c8a6fe5 · c8a6fe5
1 parent cfe3ff4
commit c8a6fe5
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git 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
@@ -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 "$@"