search

minji-o-j / YOLOv2

💻 YOLOv2 pytorch
MIT License
0 stars 0 forks source link

AssertionError: Gather function not implemented for CPU tensors #56

Closed minji-o-j closed 3 years ago

minji-o-j commented 3 years ago

버전 downgrade

pip3 install torch==1.3.1 -f https://download.pytorch.org/whl/torch_stable.html
pip3 install torchvision==0.4.1