GPU num for training

[Xiao-R-Y]

Thanks for your excellent work, I've got a problem when training with only one GPU, could you please give me some guidance on non-distributed learning commands, thank you.

THE logs are as follows: Training command is /home/zhangyang/envs/anaconda3/envs/realVSR/bin/python -m torch.distributed.launch --nproc_per_node=1 --master_port=21932 /home/zhangyang/envs/anaconda3/envs/realVSR/lib/python3.7/site-packages/mmedit/.mim/tools/train.py configs/realbasicvsr_wogan_c64b20_2x30x8_lr1e-4_300k_reds.py --launcher pytorch. /home/zhangyang/envs/anaconda3/envs/realVSR/lib/python3.7/site-packages/mmcv/init.py:21: UserWarning: On January 1, 2023, MMCV will release v2.0.0, in which it will remove components related to the training process and add a data transformation module. In addition, it will rename the package names mmcv to mmcv-lite and mmcv-full to mmcv. See https://github.com/open-mmlab/mmcv/blob/master/docs/en/compatibility.md for more details. 'On January 1, 2023, MMCV will release v2.0.0, in which it will remove ' /home/zhangyang/envs/anaconda3/envs/realVSR/lib/python3.7/site-packages/mmedit/utils/setup_env.py:33: UserWarning: Setting OMP_NUM_THREADS environment variable for each process to be 1 in default, to avoid your system being overloaded, please further tune the variable for optimal performance in your application as needed. f'Setting OMP_NUM_THREADS environment variable for each process ' /home/zhangyang/envs/anaconda3/envs/realVSR/lib/python3.7/site-packages/mmedit/utils/setup_env.py:43: UserWarning: Setting MKL_NUM_THREADS environment variable for each process to be 1 in default, to avoid your system being overloaded, please further tune the variable for optimal performance in your application as needed. f'Setting MKL_NUM_THREADS environment variable for each process ' Traceback (most recent call last): File "/home/zhangyang/envs/anaconda3/envs/realVSR/lib/python3.7/site-packages/mmedit/.mim/tools/train.py", line 171, in main() File "/home/zhangyang/envs/anaconda3/envs/realVSR/lib/python3.7/site-packages/mmedit/.mim/tools/train.py", line 108, in main cfg.dump(osp.join(cfg.work_dir, osp.basename(args.config))) File "/home/zhangyang/envs/anaconda3/envs/realVSR/lib/python3.7/site-packages/mmcv/utils/config.py", line 596, in dump f.write(self.pretty_text) File "/home/zhangyang/envs/anaconda3/envs/realVSR/lib/python3.7/site-packages/mmcv/utils/config.py", line 508, in pretty_text text, _ = FormatCode(text, style_config=yapf_style, verify=True) TypeError: FormatCode() got an unexpected keyword argument 'verify' Traceback (most recent call last): File "/home/zhangyang/envs/anaconda3/envs/realVSR/lib/python3.7/runpy.py", line 193, in _run_module_as_main "main", mod_spec) File "/home/zhangyang/envs/anaconda3/envs/realVSR/lib/python3.7/runpy.py", line 85, in _run_code exec(code, run_globals) File "/home/zhangyang/envs/anaconda3/envs/realVSR/lib/python3.7/site-packages/torch/distributed/launch.py", line 260, in main() File "/home/zhangyang/envs/anaconda3/envs/realVSR/lib/python3.7/site-packages/torch/distributed/launch.py", line 256, in main cmd=cmd) subprocess.CalledProcessError: Command '['/home/zhangyang/envs/anaconda3/envs/realVSR/bin/python', '-u', '/home/zhangyang/envs/anaconda3/envs/realVSR/lib/python3.7/site-packages/mmedit/.mim/tools/train.py', '--local_rank=0', 'configs/realbasicvsr_wogan_c64b20_2x30x8_lr1e-4_300k_reds.py', '--launcher', 'pytorch']' returned non-zero exit status 1. Traceback (most recent call last): File "/home/zhangyang/envs/anaconda3/envs/realVSR/bin/mim", line 8, in sys.exit(cli()) File "/home/zhangyang/envs/anaconda3/envs/realVSR/lib/python3.7/site-packages/click/core.py", line 1157, in call return self.main(*args, **kwargs) File "/home/zhangyang/envs/anaconda3/envs/realVSR/lib/python3.7/site-packages/click/core.py", line 1078, in main rv = self.invoke(ctx) File "/home/zhangyang/envs/anaconda3/envs/realVSR/lib/python3.7/site-packages/click/core.py", line 1688, in invoke return _process_result(sub_ctx.command.invoke(sub_ctx)) File "/home/zhangyang/envs/anaconda3/envs/realVSR/lib/python3.7/site-packages/click/core.py", line 1434, in invoke return ctx.invoke(self.callback, **ctx.params) File "/home/zhangyang/envs/anaconda3/envs/realVSR/lib/python3.7/site-packages/click/core.py", line 783, in invoke return __callback(*args, **kwargs) File "/home/zhangyang/envs/anaconda3/envs/realVSR/lib/python3.7/site-packages/mim/commands/train.py", line 111, in cli other_args=other_args) File "/home/zhangyang/envs/anaconda3/envs/realVSR/lib/python3.7/site-packages/mim/commands/train.py", line 262, in train cmd, env=dict(os.environ, MASTER_PORT=str(port))) File "/home/zhangyang/envs/anaconda3/envs/realVSR/lib/python3.7/subprocess.py", line 328, in check_call raise CalledProcessError(retcode, cmd) subprocess.CalledProcessError: Command '['/home/zhangyang/envs/anaconda3/envs/realVSR/bin/python', '-m', 'torch.distributed.launch', '--nproc_per_node=1', '--master_port=21932', '/home/zhangyang/envs/anaconda3/envs/realVSR/lib/python3.7/site-packages/mmedit/.mim/tools/train.py', 'configs/realbasicvsr_wogan_c64b20_2x30x8_lr1e-4_300k_reds.py', '--launcher', 'pytorch']' returned non-zero exit status 1.