用原版的就可以，用这个版本就一直报错 IndexError: tuple index out of range

[sctm002]

bsz = inputs[0].size(self.dim) IndexError: tuple index out of range 原版是这样写的： model = DataParallel(model, device_ids=[int(i) for i in args.device.split(',')]) 按这个版本的介绍这样写： model = BalancedDataParallel(1,model, dim=0).cuda() 就一直报错。 这个的说明内容也太少了吧。 不知道从何排错。

[sherlcok314159]

说明你将输入送进模型时，可能按照字典的形式，常见于transformers中，如：

inputs = {
    "input_ids": ...,
    "attention_mask": ...,
    "token_type_ids": ...,
}
outputs = model(**inputs)

再看下源码的处理：

def scatter(self, inputs, kwargs, device_ids):
    # 从inputs第一个输入中获取bsz
    bsz = inputs[0].size(self.dim)
    num_dev = len(self.device_ids)

所以当你上面输入过来的时候，Inputs就是个空的元组，肯定不work，可以将scatter获取bsz的代码改成我这个：

def scatter(self, inputs, kwargs, device_ids):
    if len(inputs) > 0:
        bsz = inputs[0].size(self.dim)
    elif kwargs:
        bsz = list(kwargs.values())[0].size(self.dim)
    else:
        raise ValueError("You must pass inputs to the model!")
    num_dev = len(self.device_ids)
    ...