Bug fix of AudioSeqField (OpenNMT#1319)

* Bug fix
AMusketeer · Feb 24, 2019 · cae083f · cae083f
1 parent 9865443
commit cae083f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/onmt/inputters/audio_dataset.py b/onmt/inputters/audio_dataset.py
@@ -211,7 +211,7 @@ def numericalize(self, arr, device=None):
             arr = arr.permute(3, 0, 1, 2)
         if self.sequential:
             arr = arr.contiguous()
-
+        arr = arr.to(device)
         if self.include_lengths:
             return arr, lengths
         return arr