Skip to content

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

@0xLDF

Description

@0xLDF

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

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

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions