Fix bug? (?)

data_loader.py

@@ -226,7 +226,7 @@ class FileDataset(Dataset):
         if "labels" in sample:
             # return UDP as 4chn XYZV float tensor
             sample["labels"] = torch.from_numpy(
-                sample["labels"].transpose((2, 0, 1)))
+                sample["labels"].transpose((2, 0, 1)).astype(np.float32))
             assert (sample["labels"].dtype == torch.float32)
 
         if "image_np" in sample:
@@ -270,4 +270,4 @@ class FileDataset(Dataset):
             "character_masks": character_masks
         })
         # do not make fake labels in inference
-        return sample
+        return sample