facebookresearch / EGG

EGG: Emergence of lanGuage in Games
MIT License
281 stars 99 forks source link

Putting cached lengths on CPU before calling RNN #166

Closed eugene-kharitonov closed 3 years ago

eugene-kharitonov commented 3 years ago

In pytorch 1.7 nn.utils.rnn.pack_padded_sequence crashes when passing cuda-placed lengths (before, it they were implicitly copied over).

Description

Copying over to CPU.

How Has This Been Tested?

UTs pass.