[egs] GPU id defaults to $CUDA_VISIBLE_DEVICES in all recipes (aste…

…roid-team#128)
sreyareddy15 · May 29, 2020 · 7c1cf78 · 7c1cf78
1 parent 83a6bf1
commit 7c1cf78
Showing 9 changed files with 9 additions and 9 deletions.
diff --git a/egs/dns_challenge/baseline/run.sh b/egs/dns_challenge/baseline/run.sh
@@ -16,7 +16,7 @@ python_path=python
 stage=0
 tag=""  # Controls the directory name associated to the experiment
 # You can ask for several GPUs using id (passed to CUDA_VISIBLE_DEVICES)
-id=
+id=$CUDA_VISIBLE_DEVICES
 
 # Experiment config
 is_complex=1  # If we use a complex network for training.

diff --git a/egs/fuss/baseline/run.sh b/egs/fuss/baseline/run.sh
@@ -30,7 +30,7 @@ python_path=python
 stage=0  # Controls from which stage to start
 tag=""  # Controls the directory name associated to the experiment
 # You can ask for several GPUs using id (passed to CUDA_VISIBLE_DEVICES)
-id=
+id=$CUDA_VISIBLE_DEVICES
 
 
 # Data

diff --git a/egs/librimix/ConvTasNet/run.sh b/egs/librimix/ConvTasNet/run.sh
@@ -19,7 +19,7 @@ python_path=python
 stage=0  # Controls from which stage to start
 tag=""  # Controls the directory name associated to the experiment
 # You can ask for several GPUs using id (passed to CUDA_VISIBLE_DEVICES)
-id=
+id=$CUDA_VISIBLE_DEVICES
 out_dir=librimix # Controls the directory name associated to the evaluation results inside the experiment directory
 
 # Network config

diff --git a/egs/wham/ConvTasNet/run.sh b/egs/wham/ConvTasNet/run.sh
@@ -24,7 +24,7 @@ python_path=python
 stage=0  # Controls from which stage to start
 tag=""  # Controls the directory name associated to the experiment
 # You can ask for several GPUs using id (passed to CUDA_VISIBLE_DEVICES)
-id=
+id=$CUDA_VISIBLE_DEVICES
 
 # Data
 task=sep_clean  # Specify the task here (sep_clean, sep_noisy, enh_single, enh_both)

diff --git a/egs/wham/DPRNN/run.sh b/egs/wham/DPRNN/run.sh
@@ -24,7 +24,7 @@ python_path=python
 stage=3  # Controls from which stage to start
 tag=""  # Controls the directory name associated to the experiment
 # You can ask for several GPUs using id (passed to CUDA_VISIBLE_DEVICES)
-id=0
+id=$CUDA_VISIBLE_DEVICES
 
 # Data
 task=sep_clean  # Specify the task here (sep_clean, sep_noisy, enh_single, enh_both)

diff --git a/egs/wham/DynamicMixing/run.sh b/egs/wham/DynamicMixing/run.sh
@@ -30,7 +30,7 @@ python_path=python
 stage=3  # Controls from which stage to start
 tag=""  # Controls the directory name associated to the experiment
 # You can ask for several GPUs using id (passed to CUDA_VISIBLE_DEVICES)
-id=0
+id=$CUDA_VISIBLE_DEVICES
 
 # Data
 data_dir=data  # Local data directory (No disk space needed)

diff --git a/egs/wham/TwoStep/run.sh b/egs/wham/TwoStep/run.sh
@@ -24,7 +24,7 @@ python_path=python
 stage=0  # Controls from which stage to start
 tag=""  # Controls the directory name associated to the experiment
 # You can ask for several GPUs using id (passed to CUDA_VISIBLE_DEVICES)
-id=
+id=$CUDA_VISIBLE_DEVICES
 
 # Data
 data_dir=data  # Local data directory (No disk space needed)

diff --git a/egs/whamr/TasNet/run.sh b/egs/whamr/TasNet/run.sh
@@ -24,7 +24,7 @@ python_path=python
 stage=0  # Controls from which stage to start
 tag=""  # Controls the directory name associated to the experiment
 # You can ask for several GPUs using id (passed to CUDA_VISIBLE_DEVICES)
-id=
+id=$CUDA_VISIBLE_DEVICES
 
 # Data
 task=sep_clean  # Specify the task here (sep_clean, sep_noisy, sep_reverb, sep_reverb_noisy)

diff --git a/egs/wsj0-mix/DeepClustering/run.sh b/egs/wsj0-mix/DeepClustering/run.sh
@@ -25,7 +25,7 @@ python_path=python
 stage=3  # Controls from which stage to start
 tag=""  # Controls the directory name associated to the experiment
 # You can ask for several GPUs using id (passed to CUDA_VISIBLE_DEVICES)
-id=
+id=$CUDA_VISIBLE_DEVICES
 
 # Data
 #data_dir=data  # Local data directory (No disk space needed)