From c8a6fe5bc8328388118c62a0b606097afe27e172 Mon Sep 17 00:00:00 2001
From: "pierre.delaunay" <delaunap@rtx5.server.mila.quebec>
Date: Thu, 6 Jun 2024 13:38:38 -0400
Subject: [PATCH] -

---
 benchmarks/torchvision/voirfile.py | 2 +-
 scripts/article/run_cuda.sh        | 3 +++
 2 files changed, 4 insertions(+), 1 deletion(-)

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 "$@"