search

zhengchen1999 / DAT

PyTorch code for our ICCV 2023 paper "Dual Aggregation Transformer for Image Super-Resolution"
Apache License 2.0
373 stars 34 forks source link

运行tran.py出现的问题 #51

Open sure66-sda opened 1 month ago

sure66-sda commented 1 month ago

File "D:/pycharm/PyCharm Community Edition 2022.3.3/plugins/python-ce/helpers/pydev/pydevd.py", line 1496, in _exec pydev_imports.execfile(file, globals, locals) # execute the script File "D:\pycharm\PyCharm Community Edition 2022.3.3\plugins\python-ce\helpers\pydev_pydev_imps_pydev_execfile.py", line 18, in execfile exec(compile(contents+"\n", file, 'exec'), glob, loc) File "D:\lunwen\DAT-main\basicsr\train.py", line 215, in train_pipeline(root_path) File "D:\lunwen\DAT-main\basicsr\train.py", line 93, in train_pipeline opt, args = parse_options(root_path, is_train=True) File "D:\lunwen\DAT-main\basicsr\utils\options.py", line 106, in parse_options init_dist(args.launcher) File "D:\lunwen\DAT-main\basicsr\utils\dist_util.py", line 14, in init_dist _init_dist_pytorch(backend, **kwargs) File "D:\lunwen\DAT-main\basicsr\utils\dist_util.py", line 22, in _init_dist_pytorch rank = int(os.environ['RANK']) File "D:\anaconda3\envs\lld\lib\os.py", line 675, in getitem raise KeyError(key) from None KeyError: 'RANK'

sure66-sda commented 1 month ago

出现这个问题是哪个配置文件路径没调用吗，纯新手想复现一下

zhengchen1999 commented 1 month ago

你训练的指令是什么，以及运行代码的硬件配置（GPU个数）。

sure66-sda commented 1 month ago

你训练的指令是什么，以及运行代码的硬件配置（GPU个数）。

torchrun --nproc_per_node=4 --nnodes=1 --node_rank=0 --master_addr="127.0.0.1" --master_port=29500 D:/lunwen/DAT-main/basicsr/train.py -opt D:\lunwen\DAT-main\basicsr\train_DAT_2_x2.yml gpu的话只有一个4060

zhengchen1999 commented 1 month ago

torchrun --nproc_per_node=4 修改为 torchrun --nproc_per_node=1