赞
踩
问题的提示信息:subprocess.CalledProcessError: Command ‘[’/public/home/***/anaconda3/bin/python’, ‘-u’, ‘./tools/train.py’, ‘–local_rank=7’, ‘configs/bsds/EDTER_BIMLA_320x320_80k_bsds_bs_8.py’, ‘–launcher’, ‘pytorch’]’ returned non-zero exit status 1.
在 DistributedDataParallel 中加入find_unused_parameters = True
model = torch.nn.parallel.DistributedDataParallel(model,device_ids=[args.local_rank],output_device=args.local_rank, find_unused_parameters=True)
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。