Skip to content
Unverified Commit ef1b0f08 authored by antoniogois's avatar antoniogois Committed by GitHub
Browse files

fix pack_padded_sequence()

Hi!
It seems like for pytorch 1.7.0, pack_padded_sequence's src_length  must be in cpu, even if we're using cuda: https://github.com/pytorch/pytorch/issues/43227

I also tried this command in pytorch 1.6.0 to check for backward compatibility and there it works fine, both with and without adding .cpu()
parent 4f0abcf5
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment