Thank you so much for your reply.
I fixed the error by transposing the numpy array in the Dataset’s getitem as you requested 
The line is now:
single_track = torch.from_numpy(np.transpose(genfromtxt(self.list_of_data_files[index], delimiter=','))[:,1:])

Unfortunately a new error popped up now calling an error in the type of tensor. I use FloatTensor but it expects a DoubleTensor.
RuntimeError: Expected object of type torch.DoubleTensor but found type torch.FloatTensor for argument #2 'weight'

For what i understand its an error that occurs when it uses both cpu and gpu? I don’t know how to set which processor to use so will have to look into that