torchvision-FasterRCNN

torchvision-FasterRCNN copied to clipboard

请问你的pytorch版本以及tochvision版本是多少

2

hx358031364

chunk expects at least a 1-dimensional tensor

1

在执行多GPU操作的时候报了chunk expects at least a 1-dimensional tensor，我输入的命令为python -m torch.distributed.launch --nproc_per_node=2 --use_env train.py --data-path ./CocoDataset --dataset coco --num-classes 103 --batch-size 64 --epochs 50

hx358031364