Fix cnn training notebook

notebooks/Gallery.ipynb

@@ -37,6 +37,7 @@
     "|-------|--------------|-------|-------------|-------------|\n",
     "| **Simplified weather model** | Train a reduced-size weather model on a standard GPU with fetchable dataset | ![Image showing FourCastMini prediction outputs](https://pyearthtools.readthedocs.io/en/latest/_images/notebooks_tutorial_FourCastMini_Demo_18_1.png) | [Train and run a simplified global weather model (low hardware and data requirements)](./tutorial/FourCastMini_Demo.ipynb) | 18 Aug 2025 |\n",
     "| **MLX Demo** | Shows how to integrate PyEarthTools with a non-PyTorch framework (Apple MLX) optimised for M-series chips | ![Image showing weather model outputs from MLX demo](https://pyearthtools.readthedocs.io/en/latest/_images/notebooks_tutorial_MLX-Demo-Custom-Arch_13_1.png) |  [MLX Framework Example](./tutorial/MLX-Demo-Custom-Arch.ipynb) | 8 Jun 2025 | \n",
+    "| **Convolutional Neural Net on ERA5** | Shows all steps to train a CNN on ERA5, running on CPU or a standard GPU | ![Image showing weather model outputs](https://pyearthtools.readthedocs.io/en/latest/_images/notebooks_tutorial_CNN-Model-Training_55_1.png) | [End-to-end CNN Training Example](./tutorial/CNN-Model-Training.ipynb) | 25 Aug 2025 |\n",
     "| **Radar Visualisation** | Shows how to visualise radar data as a time-series, in 2D and in 3D | ![Image showing a top down view of radar data](https://pyearthtools.readthedocs.io/en/latest/_images/notebooks_RadarVisualisation_10_1.png) | [Radar Visualisation](./RadarVisualisation.ipynb) | 23 Aug 2025 |\n"
    ]
   },
@@ -56,7 +57,6 @@
     "|       | **ENSO Forecast**: XGBoost and MLP time-series forecasting | ![Image showing plots of model accuracy](https://pyearthtools.readthedocs.io/en/latest/_images/notebooks_tutorial_ENSO_Tutorial_ENSO_Forecast_51_0.png) | [ENSO Forecast](./tutorial/ENSO_Tutorial/ENSO_Forecast.ipynb) | 16 Aug 2025 |\n",
     "|       | **ENSO Pipeline**: PyEarthTools Pipeline approaches for ENSO | ![Imagine showing time series of ENSO anomaly values](https://pyearthtools.readthedocs.io/en/latest/_images/notebooks_tutorial_ENSO_Tutorial_ENSO_Pipeline_8_1.png) | [ENSO Pipeline](./tutorial/ENSO_Tutorial/ENSO_Pipeline.ipynb) | 16 Aug 2025 |\n",
     "|       | **ENSO Gridded MLP**: Using PyEarthTools pipelines for spatial-temporal approaches to ENSO modelling | ![Image depicting the data pipeline](https://pyearthtools.readthedocs.io/en/latest/_images/notebooks_tutorial_ENSO_Tutorial_ENSO_Gridded_MLP_19_0.png) |  [ENSO Gridded MLP](./tutorial/ENSO_Tutorial/ENSO_Gridded_MLP.ipynb) | 16 Aug 2025 |\n",
-    "| **Convolutional Neural Net on ERA5** | Shows all steps to train a CNN on ERA5 | ![Image showing weather model outputs](https://pyearthtools.readthedocs.io/en/latest/_images/notebooks_tutorial_CNN-Model-Training_55_1.png) | [End-to-end CNN Training Example](./tutorial/CNN-Model-Training.ipynb) | 18 Aug 2025, minor issue at end of notebook, will investigate |\n",
     "| **Training a high resolution global atmospheric model** | Shows all steps to train the FourCastNeXt neural earth system model |  | [Training FourCastNeXt](./tutorial/FourCastNeXt_Training.ipynb) | 22 Aug 2025 |\n",
     "| **Predicting the weather** | Shows how to use a trained atmospheric model to make weather predictions using the FourCastNeXt model | ![Image showing model outputs](https://pyearthtools.readthedocs.io/en/latest/_images/notebooks_demo_FourCastNeXt_Inference_9_1.png)   | [Make a weather prediction with FourCastNeXt](./demo/FourCastNeXt_Inference.ipynb) | NOT working on 1 June 2025, requires fixes to the configuration files to work for all users, will be restored in future |\n",
     "\n"
@@ -116,14 +116,6 @@
     "| Modifications | Introduction to pipeline modifications | [Pipeline Modifications](./pipeline/Modifications.ipynb) | 22 Aug 2025 |\n",
     "| Branching | -- | [Pipeline Branching](./pipeline/Branching.ipynb) | 18 Aug 2025 |\n"
    ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "77a95040-0767-4b0f-8549-128c3286fc11",
-   "metadata": {},
-   "outputs": [],
-   "source": []
   }
  ],
  "metadata": {