mmdetection-mini icon indicating copy to clipboard operation
mmdetection-mini copied to clipboard

Published 20 hours ago • verified publisher icon hhaAndroid

cv_core/parallel/data_parallel.py中的53行和77行是不是少写了[0]

Open Bilibilee opened this issue 3 years ago • 0 comments

作者您好，在阅读你的MMDataParallel源码时发现，53行和77行与mmcv中不符。 你的代码库是return self.module.train_step(*inputs, **kwargs) mmcv代码库是return self.module.train_step(*inputs[0], **kwargs[0])

scatter返回的tuple嘛，应该是需要[0]取一下元素的

Bilibilee avatar Dec 26 '21 13:12 Bilibilee