pytorch0.4+cuda10.0+py3.6.5
代码:https://github.com/amdegroot/ssd.pytorch
问题1:RuntimeError: The shape of the mask [32, 8732] at index 0 does not match the shape of the indexed tensor [279424, 1] at index 0
解决:
修改multibox_loss.py
step1: switch the two lines 97,98:
loss_c = loss_c.view(num, -1)
loss_c[pos] = 0 # filter out pos boxes for now
step2: change the line144 N = num_pos.data.sum() to
N = num_pos.data.sum().double()
loss_l = loss_l.double()/N
loss_c = loss_c.double()/N
问题2:自动跳出迭代:
需要修改3个文件:detection.py; multibox_loss.py; train.py
https://github.com/kentaroy47/ssd.pytorch/commit/4dc4bcead9b5f2d12108a935c85a87dfca83fe82
在运行到3670 后,仍然跳出迭代
再次修改:train.py
batch_iterator = None
for iterat