diff --git a/aimodel/src/deeplabv3_plus_test_rainfall.py b/aimodel/src/deeplabv3_plus_test_rainfall.py
index f0880c5..8b35dfb 100755
--- a/aimodel/src/deeplabv3_plus_test_rainfall.py
+++ b/aimodel/src/deeplabv3_plus_test_rainfall.py
@@ -262,7 +262,7 @@ def plot_predictions(filepath, input_items, colormap, model):
 		# label_colourmap = decode_segmentation_masks(input_pair[1], colormap, 2)
 		prediction_colormap = decode_segmentation_masks(prediction_mask, colormap, 2)
 		
-		print("DEBUG:plot_predictions INFER", prediction_mask)
+		print("DEBUG:plot_predictions INFER", str(prediction_mask.numpy().tolist()).replace("], [", "],\n["))
 		
 		plot_samples_matplotlib(
 			filepath.replace("$$", str(i)),