diff --git a/modules/deeplearning/icing_fcn.py b/modules/deeplearning/icing_fcn.py index 32d00b884878b653eba4fcee9a867d2edc069aa6..950c9715735ec5a76ed8760b4afbf70b1e3557b9 100644 --- a/modules/deeplearning/icing_fcn.py +++ b/modules/deeplearning/icing_fcn.py @@ -976,6 +976,7 @@ class IcingIntensityFCN: print(data[0].shape, data[1].shape) pred = self.model([data]) print(pred.shape, np.histogram(pred.numpy())) + pred_s.append(pred) preds = np.concatenate(pred_s) preds = preds[:,0]