update
This commit is contained in:
parent
3b46081df4
commit
b37b513a6c
@ -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) \
|
||||
|
Loading…
x
Reference in New Issue
Block a user