RuntimeError: The shape of the mask [2, 128, 128] at index 0 does not match the shape of the indexed tensor [1, 2, 2, 128] at index 0

princeton-vl / CornerNet

BSD 3-Clause "New" or "Revised" License

2.36k stars 475 forks source link

RuntimeError: The shape of the mask [2, 128, 128] at index 0 does not match the shape of the indexed tensor [1, 2, 2, 128] at index 0 #154

Open duanqipeng opened 4 years ago

duanqipeng commented 4 years ago

i got a problem with RuntimeError: The shape of the mask [2, 128, 128] at index 0 does not match the shape of the indexed tensor [1, 2, 2, 128] at index 0

use four gpu