Skip to content

Run on the PyTorch 1.3+ #5

@xw-hu

Description

@xw-hu
  1. nets.py: return irnn()(input, self.up_weight.weight....) -> return irnn().forward(input, self.up_weight.weight....)
  2. irnn.py: @cupy.util.memorize(...) -> @cupy.memorize(...)

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