Closed fygfff closed 6 months ago
File "G:\yanjiu\lunwen1\DUO\daima\MotionRGBD-PAMI-main\train_fusion.py", line 690, in main(args) File "G:\yanjiu\lunwen1\DUO\daima\MotionRGBD-PAMI-main\train_fusion.py", line 283, in main model = nn.parallel.DistributedDataParallel(model, device_ids=[args.gpu], find_unused_parameters=False) File "E:\anaconda\envs\mrgbd\lib\site-packages\torch\nn\parallel\distributed.py", line 593, in init
Hi, please try setting find_unused_parameters=True
.
File "G:\yanjiu\lunwen1\DUO\daima\MotionRGBD-PAMI-main\train_fusion.py", line 690, in
main(args)
File "G:\yanjiu\lunwen1\DUO\daima\MotionRGBD-PAMI-main\train_fusion.py", line 283, in main
model = nn.parallel.DistributedDataParallel(model, device_ids=[args.gpu], find_unused_parameters=False)
File "E:\anaconda\envs\mrgbd\lib\site-packages\torch\nn\parallel\distributed.py", line 593, in init