torchvision-FasterRCNN icon indicating copy to clipboard operation
torchvision-FasterRCNN copied to clipboard

Published 20 hours ago • verified publisher icon ouening

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

Open hx358031364 opened this issue 4 years ago • 2 comments

hx358031364 avatar Dec 23 '20 10:12 hx358031364

这个版本区别应该没影响吧，我最近用最新的pytorch1.7.1，torchvision0.8.2也能跑通

ouening avatar Dec 23 '20 10:12 ouening

这个版本区别应该没影响吧，我最近用最新的pytorch1.7.1，torchvision0.8.2也能跑通

哦哦，我在尝试多GPU的训练，可能回跟版本有关系吧，data_loader读数据返回后，输入模型中的时候总是报chunk expects at least a 1-dimensional tensor错误 我看返回的images，和targets都有值

hx358031364 avatar Dec 23 '20 11:12 hx358031364