From 1f4136ee723a457e4f4f8bcc1e5cc328881b00d3 Mon Sep 17 00:00:00 2001 From: Piotr Kozakowski Date: Fri, 3 Nov 2017 18:40:07 +0100 Subject: Add option to run without CUDA --- dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'dataset.py') diff --git a/dataset.py b/dataset.py index 3bb035f..c461330 100644 --- a/dataset.py +++ b/dataset.py @@ -58,7 +58,7 @@ class DataLoader(DataLoaderBase): to_index = seq_begin + self.seq_len sequences = batch[:, from_index : to_index] input_sequences = sequences[:, : -1] - target_sequences = sequences[:, self.overlap_len :] + target_sequences = sequences[:, self.overlap_len :].contiguous() yield (input_sequences, reset, target_sequences) -- cgit v1.2.3-70-g09d2