Update docker_run.sh

457550aa · yilun_jin · 656eb923 · 457550aa
Commit 457550aa authored 11 months ago by yilun_jin
--- a/docker_run.sh
+++ b/docker_run.sh
@@ -35,6 +35,7 @@ docker run \
    --gpus all \
    -v "$(pwd)":/submission \
    -w /submission \
+    --shm-size=10.24gb\
    $IMAGE_NAME python local_evaluation.py

 # Note: We assume you have nvidia-container-toolkit installed and configured