I create a model that returns a dictionary. However, when I try to train it with DataParallel, it seems like the output dictionary cannot be properly gathered.
line 62, in gather_map
    return type(out)(map(gather_map, zip(*outputs)))
TypeError: zip argument #1 must support iteration