Convert to torch tensor.

climax/core/predict.py

@@ -64,7 +64,9 @@ def predict_ERA5(net, ERA5_ds, predictand, batch_size=16, **kwargs):
     elif predictand == 'pr':
         ds = {
             # probability of precipitation
-            'prob': torch.sigmoid(target[:, 0, ...].squeeze()),
+            'prob': torch.sigmoid(
+                torch.as_tensor(target[:, 0, ...].squeeze(),
+                                dtype=torch.float32)).numpy(),
 
             # amount of precipitation: expected value of gamma distribution
             # pr = shape * scale