set num_gpu to auto in options

2021-09-17 10:07:09 +08:00
parent 90ddf13b5e
commit 0ad2e9c61e
4 changed files with 4 additions and 4 deletions
--- a/options/train_realesrgan_x2plus.yml
+++ b/options/train_realesrgan_x2plus.yml
@@ -2,7 +2,7 @@
 name: train_RealESRGANx2plus_400k_B12G4
 model_type: RealESRGANModel
 scale: 2
-num_gpu: 4
+num_gpu: auto  # auto: can infer from your visible devices automatically. official: 4 GPUs
 manual_seed: 0

 # ----------------- options for synthesizing training data in RealESRGANModel ----------------- #
--- a/options/train_realesrgan_x4plus.yml
+++ b/options/train_realesrgan_x4plus.yml
@@ -2,7 +2,7 @@
 name: train_RealESRGANx4plus_400k_B12G4
 model_type: RealESRGANModel
 scale: 4
-num_gpu: 4
+num_gpu: auto  # auto: can infer from your visible devices automatically. official: 4 GPUs
 manual_seed: 0

 # ----------------- options for synthesizing training data in RealESRGANModel ----------------- #
--- a/options/train_realesrnet_x2plus.yml
+++ b/options/train_realesrnet_x2plus.yml
@@ -2,7 +2,7 @@
 name: train_RealESRNetx2plus_1000k_B12G4
 model_type: RealESRNetModel
 scale: 2
-num_gpu: 4
+num_gpu: auto  # auto: can infer from your visible devices automatically. official: 4 GPUs
 manual_seed: 0

 # ----------------- options for synthesizing training data in RealESRNetModel ----------------- #
--- a/options/train_realesrnet_x4plus.yml
+++ b/options/train_realesrnet_x4plus.yml
@@ -2,7 +2,7 @@
 name: train_RealESRNetx4plus_1000k_B12G4
 model_type: RealESRNetModel
 scale: 4
-num_gpu: 4
+num_gpu: auto  # auto: can infer from your visible devices automatically. official: 4 GPUs
 manual_seed: 0

 # ----------------- options for synthesizing training data in RealESRNetModel ----------------- #