From b37b513a6c74e6ed466e4fbe4aceaba0a0c25dea Mon Sep 17 00:00:00 2001
From: wataru <wataru@fdev.local.com>
Date: Sun, 11 Dec 2022 05:26:21 +0900
Subject: [PATCH] update

---
 start2.sh | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/start2.sh b/start2.sh
index c1e0b3ea..43bcc4d2 100644
--- a/start2.sh
+++ b/start2.sh
@@ -34,7 +34,7 @@ USE_GPU=${USE_GPU:-${DEFAULT_USE_GPU}}
 if [ "${MODE}" = "TRAIN" ]; then
     echo "トレーニングを開始します"
 
-    docker run -it --gpus all --shm-size=128M \
+    docker run -it --rm --gpus all --shm-size=128M \
         -v `pwd`/work_dir/logs:/MMVC_Trainer/logs \
         -v `pwd`/work_dir/dataset:/MMVC_Trainer/dataset \
         -v `pwd`/work_dir/info:/MMVC_Trainer/info \
@@ -50,7 +50,7 @@ elif [ "${MODE}" = "MMVC" ]; then
     if [ "${USE_GPU}" = "on" ]; then
         echo "MMVCを起動します(with gpu)"
 
-        docker run -it --gpus all --shm-size=128M \
+        docker run -it --rm --gpus all --shm-size=128M \
         -v `pwd`/vc_resources:/resources \
         -e LOCAL_UID=$(id -u $USER) \
         -e LOCAL_GID=$(id -g $USER) \
@@ -60,7 +60,7 @@ elif [ "${MODE}" = "MMVC" ]; then
         $DOCKER_IMAGE "$@"
     else
         echo "MMVCを起動します(only cpu)"
-        docker run -it --shm-size=128M \
+        docker run -it --rm --shm-size=128M \
         -v `pwd`/vc_resources:/resources \
         -e LOCAL_UID=$(id -u $USER) \
         -e LOCAL_GID=$(id -g $USER) \