awslabs / sockeye

Sequence-to-sequence framework with a focus on Neural Machine Translation based on PyTorch
https://awslabs.github.io/sockeye/
Apache License 2.0
1.21k stars 323 forks source link

Fixed GPU-based scoring by copying to cpu tensor first before converting to numpy. #1020

Closed fhieber closed 2 years ago

fhieber commented 2 years ago

Pull Request Checklist

By submitting this pull request, I confirm that my contribution is made under the terms of the Apache 2.0 license.

ehasler commented 2 years ago

looks good