Skip to content

latest version of pytorch requires index to be Variable in index_select() #6

@shirishr

Description

@shirishr

I think
index=..... should be wrapped in as index=Variable(.....) in Decoder class as:
masked = masked.index_select(dim=0, index=Variable(max_length_indices.squeeze(1).data))

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