diff --git a/pysegcnn/core/dataset.py b/pysegcnn/core/dataset.py
index 00b3fdae36360fe6d934facfe3f864e9e7fe05d0..ba9df9635831fec44bd81ce803c6d92147299a20 100644
--- a/pysegcnn/core/dataset.py
+++ b/pysegcnn/core/dataset.py
@@ -346,7 +346,7 @@ class ImageDataset(Dataset):
             x, y = scene['transform'](x, y)
 
         # convert to torch tensors
-        x = self.to_tensor(x, dtype=torch.float16)
+        x = self.to_tensor(x, dtype=torch.float32)
         y = self.to_tensor(y, dtype=torch.uint8)
 
         return x, y