exp文件中将self.mosaic_prob置为0.5时报错 'int' object has no attribute 'numel'

[1VeniVediVeci1]

trafficstars

训练nano网络的时候，exps/default/nano.py里有一句self.mosaic_prob = 0.5，加上后训练时就会报如下图所示错误

[jhl13]

Same problem.

[jhl13]

#658 solve the problem.

[txx-899]

我也遇到了这个问题，最新代码里已经修改了此处，但还是会报错

[Single430]

Same problem

[Single430]

Same problem

solve the problem

[sean-wade]

同样的问题，batch=1时不会报错，猜测是batch 拼接的时候出的问题？

[rimq14]

同样的问题，有人解决了吗？