search

clay-lab / transductions

A PyTorch framework for creating, running, and reproducing experiments on seq2seq models.
6 stars 3 forks source link

Add cuda magic to GRUTridentDecoder #25

Open NoahAmsel opened 4 years ago

NoahAmsel commented 4 years ago

https://github.com/clay-lab/transductions/blob/TreeAttn/models.py#L519

Make sure that we create tensors on the right device