Skip to content

DataParallel does not support dicts as first (or sole) input #1857

New issue
New issue
Closed
Closed
DataParallel does not support dicts as first (or sole) input#1857
@arunmallya

Description

@arunmallya
arunmallya
opened on Jun 20, 2017

pytorch/torch/nn/parallel/parallel_apply.py

Lines 21 to 22 in d9d50f8

while not isinstance(var_input, Variable):
var_input = var_input[0]
is the culprit. Assumes that the first input is a variable or a sequence containing a variable in index 0 (recursively indexed).

Workaround is to pass a dummy variable as first input, as suggested in https://pytorch.slack.com/archives/C2077MFDL/p1497986173722359

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions