lucasrufkahr/Real-ESRGAN
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

set num_gpu to auto in options

Browse Source
This commit is contained in:
Xintao
2021-09-17 10:07:09 +08:00
parent 90ddf13b5e
commit 0ad2e9c61e
4 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
options/train_realesrgan_x4plus.yml
View File

@@ -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 ----------------- #