Update README.md
Browse files
README.md
CHANGED
|
@@ -41,9 +41,12 @@ the [Gravity Wave Parameterization dataset](https://huggingface.co/datasets/Prit
|
|
| 41 |
Code for fine-tuning is available through [Github](https://github.com/NASA-IMPACT/gravity-wave-finetuning).
|
| 42 |
|
| 43 |
### Results
|
|
|
|
|
|
|
| 44 |
For the Andes (mountain waves) and the Southern Ocean (non-mountain waves),
|
| 45 |
the fine-tuned model achieves correlation coefficients of 0.99 and 0.97, respectively, when compared to the observed fluxes.
|
| 46 |
|
|
|
|
| 47 |
### Inference and demo
|
| 48 |
The github repo includes an inference script that allows to run
|
| 49 |
the [gravity_wave_model](https://huggingface.co/Prithvi-WxC/Gravity_wave_Parameterization/blob/main/magnet-flux-uvtp122-epoch-99-loss-0.1022.pt) model
|
|
|
|
| 41 |
Code for fine-tuning is available through [Github](https://github.com/NASA-IMPACT/gravity-wave-finetuning).
|
| 42 |
|
| 43 |
### Results
|
| 44 |
+
<img src="https://cdn-uploads.huggingface.co/production/uploads/6488f1d3e22a0081a561ec8f/Vk1EKgzf_j90ZPiw2hGHE.png" alt="Gravity Wave" width="1024"/>
|
| 45 |
+
|
| 46 |
For the Andes (mountain waves) and the Southern Ocean (non-mountain waves),
|
| 47 |
the fine-tuned model achieves correlation coefficients of 0.99 and 0.97, respectively, when compared to the observed fluxes.
|
| 48 |
|
| 49 |
+
|
| 50 |
### Inference and demo
|
| 51 |
The github repo includes an inference script that allows to run
|
| 52 |
the [gravity_wave_model](https://huggingface.co/Prithvi-WxC/Gravity_wave_Parameterization/blob/main/magnet-flux-uvtp122-epoch-99-loss-0.1022.pt) model
|