echo message on console before running autogluon MM-quickstart notebook

Message will be printed on the console for user to inform that autogluon multimodel-quick-start.ipynb only supports single GPU.
aws · Dec 3, 2024 · 3e3ecb2 · 3e3ecb2
1 parent 04d11b2
commit 3e3ecb2
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/test/test_artifacts/v0/scripts/run_autogluon_tests.sh b/test/test_artifacts/v0/scripts/run_autogluon_tests.sh
@@ -13,5 +13,6 @@ ret=$?
 
 if [ $ret -eq 0 ]
 then
+    echo "This notebook only supports a single GPU setup. Running on multi-GPU systems may lead to unexpected errors"
     jupyter nbconvert --execute --to python docs/tutorials/multimodal/multimodal_prediction/multimodal-quick-start.ipynb
 fi