From 0ad2e9c61e158883353ebf99ff5392e31d760c8d Mon Sep 17 00:00:00 2001 From: Xintao Date: Fri, 17 Sep 2021 10:07:09 +0800 Subject: [PATCH] set num_gpu to auto in options --- options/train_realesrgan_x2plus.yml | 2 +- options/train_realesrgan_x4plus.yml | 2 +- options/train_realesrnet_x2plus.yml | 2 +- options/train_realesrnet_x4plus.yml | 2 +- 4 files changed, 4 insertions(+), 4 deletions(-) diff --git a/options/train_realesrgan_x2plus.yml b/options/train_realesrgan_x2plus.yml index b0c213e..7573d7f 100644 --- 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 ----------------- # diff --git a/options/train_realesrgan_x4plus.yml b/options/train_realesrgan_x4plus.yml index cd50c0d..f4b768a 100644 --- 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 ----------------- # diff --git a/options/train_realesrnet_x2plus.yml b/options/train_realesrnet_x2plus.yml index 520132f..a3838a9 100644 --- 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 ----------------- # diff --git a/options/train_realesrnet_x4plus.yml b/options/train_realesrnet_x4plus.yml index dce0fb2..81bf156 100644 --- 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 ----------------- #