From 457550aad5358c4e049849e9fa0c6d8dd1e9898e Mon Sep 17 00:00:00 2001
From: yilun_jin <jyl.jal123@gmail.com>
Date: Sat, 25 May 2024 13:25:54 +0000
Subject: [PATCH] Update docker_run.sh
---
docker_run.sh | 1 +
1 file changed, 1 insertion(+)
diff --git a/docker_run.sh b/docker_run.sh
index 341c6cf..23dd1a6 100755
--- 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
--
GitLab