search

princeton-vl / CornerNet

BSD 3-Clause "New" or "Revised" License
2.36k stars 475 forks source link

RuntimeError: given chunk sizes don't sum up to the tensor's size (sum(chunk_sizes) == 49, but expected 1) (scatter at torch/csrc/cuda/comm.cpp:135) #61

Closed chenglong0313 closed 5 years ago

chenglong0313 commented 5 years ago

RuntimeError: given chunk sizes don't sum up to the tensor's size (sum(chunk_sizes) == 49, but expected 1) (scatter at torch/csrc/cuda/comm.cpp:135) frame #0: + 0xc4968b (0x7f0ec674c68b in /home/gong/anaconda3/lib/python3.5/site-packages/torch/_C.cpython-35m-x86_64-linux-gnu.so) frame #1: + 0x39124b (0x7f0ec5e9424b in /home/gong/anaconda3/lib/python3.5/site-packages/torch/_C.cpython-35m-x86_64-linux-gnu.so)

frame #11: THPFunction_apply(_object*, _object*) + 0x38f (0x7f0ec62726af in /home/gong/anaconda3/lib/python3.5/site-packages/torch/_C.cpython-35m-x86_64-linux-gnu.so)
chenglong0313 commented 5 years ago

sorry i modify batch_size but not modify chunk_size :)