# Skillful Nowcasting with Deep Generative Model of Radar (DGMR)
<!-- ALL-CONTRIBUTORS-BADGE:START - Do not remove or modify this section -->
[](#contributors-)
<!-- ALL-CONTRIBUTORS-BADGE:END -->
Implementation of DeepMind's Skillful Nowcasting GAN Deep Generative Model of Radar (DGMR) (https://arxiv.org/abs/2104.00954) in PyTorch Lightning.
This implementation matches as much as possible the pseudocode released by DeepMind. Each of the components (Sampler, Context conditioning stack, Latent conditioning stack, Discriminator, and Generator) are normal PyTorch modules. As the model training is a bit complicated, the overall architecture is wrapped in PyTorch Lightning.
The default parameters match what is written in the paper.
## Installation
Clone the repository, then run
```shell
pip install -r requirements.txt
pip install -e .
````
Alternatively, you can also install through ```pip install dgmr```
## Training Data
The open-sourced UK training dataset has been mirrored to [HuggingFace Datasets!](https://huggingface.co/datasets/openclimatefix/nimrod-uk-1km) This should enable training the original architecture on the original data for reproducing the results from the paper. The full dataset is roughly 1TB in size, and unfortunately, streaming the data from HF Datasets doesn't seem to work, so it has to be cached locally. We have added the sample dataset as well though, which can be directly streamed from GCP without costs.
The dataset can be loaded with
```python
from datasets import load_dataset
dataset = load_dataset("openclimatefix/nimrod-uk-1km")
```
For now, only the sample dataset support streaming in, as its data files are hosted on GCP, not HF, so it can be used with:
```python
from datasets import load_dataset
dataset = load_dataset("openclimatefix/nimrod-uk-1km", "sample", streaming=True)
```
The authors also used [MRMS](https://www.nssl.noaa.gov/projects/mrms/) US precipitation radar data as another comparison. While that dataset was not released, the MRMS data is publicly available, and we have made that data available on HuggingFace Datasets as well [here](https://huggingface.co/datasets/openclimatefix/mrms). This dataset is the raw 3500x7000 contiguous US MRMS data for 2016 through May 2022, is a few hundred GBs in size, with sporadic updates to more recent data planned. This dataset is in Zarr format, and can be streamed without caching locally through
```python
from datasets import load_dataset
dataset = load_dataset("openclimatefix/mrms", "default_sequence", streaming=True)
```
This steams the data with 24 timesteps per example, just like the UK DGMR dataset. To get individual MRMS frames, instead of a sequence, this can be achieved through
```python
from datasets import load_dataset
dataset = load_dataset("openclimatefix/mrms", "default", streaming=True)
```
## Pretrained Weights
Pretrained weights are be available through [HuggingFace Hub](https://huggingface.co/openclimatefix), currently weights trained on the sample dataset. The whole DGMR model or different components can be loaded as the following:
```python
from dgmr import DGMR, Sampler, Generator, Discriminator, LatentConditioningStack, ContextConditioningStack
model = DGMR.from_pretrained("openclimatefix/dgmr")
sampler = Sampler.from_pretrained("openclimatefix/dgmr-sampler")
discriminator = Discriminator.from_pretrained("openclimatefix/dgmr-discriminator")
latent_stack = LatentConditioningStack.from_pretrained("openclimatefix/dgmr-latent-conditioning-stack")
context_stack = ContextConditioningStack.from_pretrained("openclimatefix/dgmr-context-conditioning-stack")
generator = Generator(conditioning_stack=context_stack, latent_stack=latent_stack, sampler=sampler)
```
## Example Usage
```python
from dgmr import DGMR
import torch.nn.functional as F
import torch
model = DGMR(
forecast_steps=4,
input_channels=1,
output_shape=128,
latent_channels=384,
context_channels=192,
num_samples=3,
)
x = torch.rand((2, 4, 1, 128, 128))
out = model(x)
y = torch.rand((2, 4, 1, 128, 128))
loss = F.mse_loss(y, out)
loss.backward()
```
## Citation
```
@article{ravuris2021skillful,
author={Suman Ravuri and Karel Lenc and Matthew Willson and Dmitry Kangin and Remi Lam and Piotr Mirowski and Megan Fitzsimons and Maria Athanassiadou and Sheleem Kashem and Sam Madge and Rachel Prudden Amol Mandhane and Aidan Clark and Andrew Brock and Karen Simonyan and Raia Hadsell and Niall Robinson Ellen Clancy and Alberto Arribas† and Shakir Mohamed},
title={Skillful Precipitation Nowcasting using Deep Generative Models of Radar},
journal={Nature},
volume={597},
pages={672--677},
year={2021}
}
```
## Contributors ✨
Thanks goes to these wonderful people ([emoji key](https://allcontributors.org/docs/en/emoji-key)):
<!-- ALL-CONTRIBUTORS-LIST:START - Do not remove or modify this section -->
<!-- prettier-ignore-start -->
<!-- markdownlint-disable -->
<table>
<tbody>
<tr>
<td align="center" valign="top" width="14.28%"><a href="https://www.jacobbieker.com"><img src="https://avatars.githubusercontent.com/u/7170359?v=4?s=100" width="100px;" alt="Jacob Bieker"/><br /><sub><b>Jacob Bieker</b></sub></a><br /><a href="https://github.com/openclimatefix/skillful_nowcasting/commits?author=jacobbieker" title="Code">💻</a></td>
<td align="center" valign="top" width="14.28%"><a href="http://johmathe.name/"><img src="https://avatars.githubusercontent.com/u/467643?v=4?s=100" width="100px;" alt="Johan Mathe"/><br /><sub><b>Johan Mathe</b></sub></a><br /><a href="https://github.com/openclimatefix/skillful_nowcasting/commits?author=johmathe" title="Code">💻</a></td>
<td align="center" valign="top" width="14.28%"><a href="https://github.com/ZHANGZ1YUE"><img src="https://avatars.githubusercontent.com/u/93907996?v=4?s=100" width="100px;" alt="Z1YUE"/><br /><sub><b>Z1YUE</b></sub></a><br /><a href="https://github.com/openclimatefix/skillful_nowcasting/issues?q=author%3AZHANGZ1YUE" title="Bug reports">🐛</a></td>
<td align="center" valign="top" width="14.28%"><a href="https://github.com/GreenLimeSia"><img src="https://avatars.githubusercontent.com/u/28706611?v=4?s=100" width="100px;" alt="Nan.Y"/><br /><sub><b>Nan.Y</b></sub></a><br /><a href="#question-GreenLimeSia" title="Answering Questions">💬</a></td>
<td align="center" valign="top" width="14.28%"><a href="https://github.com/hedaobaishui"><img src="https://avatars.githubusercontent.com/u/20534146?v=4?s=100" width="100px;" alt="Taisanai"/><br /><sub><b>Taisanai</b></sub></a><br /><a href="#question-hedaobaishui" title="Answering Questions">💬</a></td>
<td align="center" valign="top" width="14.28%"><a href="https://github.com/primeoc"><img src="https://avatars.githubusercontent.com/u/75205487?v=4?s=100" width="100px;" alt="cameron"/><br /><sub><b>cameron</b></sub></a><br /><a href="#question-primeoc" title="Answering Questions">💬</a></td>
<td align="center" valign="top" width="14.28%"><a href="https://github.com/zhrli"><img src="https://avatars.githubusercontent.com/u/11074703?v=4?s=100" width="100px;" alt="zhrli"/><br /><sub><b>zhrli</b></sub></a><br /><a href="#question-zhrli" title="Answering Questions">💬</a></td>
</tr>
<tr>
<td align="center" valign="top" width="14.28%"><a href="https://github.com/najeeb-kazmi"><img src="https://avatars.githubusercontent.com/u/14131235?v=4?s=100" width="100px;" alt="Najeeb Kazmi"/><br /><sub><b>Najeeb Kazmi</b></sub></a><br /><a href="#question-najeeb-kazmi" title="Answering Questions">💬</a></td>
<td align="center" valign="top" width="14.28%"><a href="https://github.com/TQRTQ"><img src="https://avatars.githubusercontent.com/u/29155385?v=4?s=100" width="100px;" alt="TQRTQ"/><br /><sub><b>TQRTQ</b></sub></a><br /><a href="#question-TQRTQ" title="Answering Questions">💬</a></td>
<td align="center" valign="top" width="14.28%"><a href="https://www.linkedin.com/in/viktor-bordiuzha-93b078211"><img src="https://avatars.githubusercontent.com/u/43813476?v=4?s=100" width="100px;" alt="Viktor Bordiuzha"/><br /><sub><b>Viktor Bordiuzha</b></sub></a><br /><a href="#example-victor30608" title="Examples">💡</a></td>
<td align="center" valign="top" width="14.28%"><a href="https://github.com/agijsberts"><img src="https://avatars.githubusercontent.com/u/1579083?v=4?s=100" width="100px;" alt="agijsberts"/><br /><sub><b>agijsberts</b></sub></a><br /><a href="https://github.com/openclimatefix/skillful_nowcasting/commits?author=agijsberts" title="Code">💻</a></td>
<td align="center" valign="top" width="14.28%"><a href="https://github.com/Mews"><img src="https://avatars.githubusercontent.com/u/60406199?v=4?s=100" width="100px;" alt="Mews"/><br /><sub><b>Mews</b></sub></a><br /><a href="https://github.com/openclimatefix/skillful_nowcasting/commits?author=Mews" title="Tests">⚠️</a></td>
<td align="center" valign="top" width="14.28%"><a href="https://github.com/rutkovskii"><img src="https://avatars.githubusercontent.com/u/70371908?v=4?s=100" width="100px;" alt="Aleksei Rutkovskii"/><br /><sub><b>Aleksei Rutkovskii</b></sub></a><br /><a href="https://github.com/openclimatefix/skillful_nowcasting/commits?author=rutkovskii" title="Code">💻</a></td>
</tr>
</tbody>
</table>
<!-- markdownlint-restore -->
<!-- prettier-ignore-end -->
<!-- ALL-CONTRIBUTORS-LIST:END -->
This project follows the [all-contributors](https://github.com/all-contributors/all-contributors) specification. Contributions of any kind welcome!
Raw data
{
"_id": null,
"home_page": "https://github.com/openclimatefix/skillful_nowcasting",
"name": "dgmr",
"maintainer": null,
"docs_url": null,
"requires_python": null,
"maintainer_email": null,
"keywords": "artificial intelligence, deep learning, transformer, attention mechanism, metnet, forecasting, remote-sensing, gan",
"author": "Jacob Prince-Bieker",
"author_email": "jacob@bieker.tech",
"download_url": "https://files.pythonhosted.org/packages/b2/e3/bdb45e568073af2a38c90aac12ddc390d241cb5ac5bb25d51bbf5abc0bcc/dgmr-1.4.0.tar.gz",
"platform": null,
"description": "# Skillful Nowcasting with Deep Generative Model of Radar (DGMR)\n<!-- ALL-CONTRIBUTORS-BADGE:START - Do not remove or modify this section -->\n[](#contributors-)\n<!-- ALL-CONTRIBUTORS-BADGE:END -->\nImplementation of DeepMind's Skillful Nowcasting GAN Deep Generative Model of Radar (DGMR) (https://arxiv.org/abs/2104.00954) in PyTorch Lightning.\n\nThis implementation matches as much as possible the pseudocode released by DeepMind. Each of the components (Sampler, Context conditioning stack, Latent conditioning stack, Discriminator, and Generator) are normal PyTorch modules. As the model training is a bit complicated, the overall architecture is wrapped in PyTorch Lightning.\n\nThe default parameters match what is written in the paper.\n\n## Installation\n\nClone the repository, then run\n```shell\npip install -r requirements.txt\npip install -e .\n````\n\nAlternatively, you can also install through ```pip install dgmr```\n\n## Training Data\n\nThe open-sourced UK training dataset has been mirrored to [HuggingFace Datasets!](https://huggingface.co/datasets/openclimatefix/nimrod-uk-1km) This should enable training the original architecture on the original data for reproducing the results from the paper. The full dataset is roughly 1TB in size, and unfortunately, streaming the data from HF Datasets doesn't seem to work, so it has to be cached locally. We have added the sample dataset as well though, which can be directly streamed from GCP without costs.\n\nThe dataset can be loaded with\n\n```python\nfrom datasets import load_dataset\n\ndataset = load_dataset(\"openclimatefix/nimrod-uk-1km\")\n```\n\nFor now, only the sample dataset support streaming in, as its data files are hosted on GCP, not HF, so it can be used with:\n\n```python\nfrom datasets import load_dataset\n\ndataset = load_dataset(\"openclimatefix/nimrod-uk-1km\", \"sample\", streaming=True)\n```\n\nThe authors also used [MRMS](https://www.nssl.noaa.gov/projects/mrms/) US precipitation radar data as another comparison. While that dataset was not released, the MRMS data is publicly available, and we have made that data available on HuggingFace Datasets as well [here](https://huggingface.co/datasets/openclimatefix/mrms). This dataset is the raw 3500x7000 contiguous US MRMS data for 2016 through May 2022, is a few hundred GBs in size, with sporadic updates to more recent data planned. This dataset is in Zarr format, and can be streamed without caching locally through\n\n```python\nfrom datasets import load_dataset\n\ndataset = load_dataset(\"openclimatefix/mrms\", \"default_sequence\", streaming=True)\n```\n\nThis steams the data with 24 timesteps per example, just like the UK DGMR dataset. To get individual MRMS frames, instead of a sequence, this can be achieved through\n\n```python\nfrom datasets import load_dataset\n\ndataset = load_dataset(\"openclimatefix/mrms\", \"default\", streaming=True)\n```\n\n## Pretrained Weights\n\nPretrained weights are be available through [HuggingFace Hub](https://huggingface.co/openclimatefix), currently weights trained on the sample dataset. The whole DGMR model or different components can be loaded as the following:\n\n```python\nfrom dgmr import DGMR, Sampler, Generator, Discriminator, LatentConditioningStack, ContextConditioningStack\nmodel = DGMR.from_pretrained(\"openclimatefix/dgmr\")\nsampler = Sampler.from_pretrained(\"openclimatefix/dgmr-sampler\")\ndiscriminator = Discriminator.from_pretrained(\"openclimatefix/dgmr-discriminator\")\nlatent_stack = LatentConditioningStack.from_pretrained(\"openclimatefix/dgmr-latent-conditioning-stack\")\ncontext_stack = ContextConditioningStack.from_pretrained(\"openclimatefix/dgmr-context-conditioning-stack\")\ngenerator = Generator(conditioning_stack=context_stack, latent_stack=latent_stack, sampler=sampler)\n```\n\n## Example Usage\n\n```python\nfrom dgmr import DGMR\nimport torch.nn.functional as F\nimport torch\n\nmodel = DGMR(\n forecast_steps=4,\n input_channels=1,\n output_shape=128,\n latent_channels=384,\n context_channels=192,\n num_samples=3,\n )\nx = torch.rand((2, 4, 1, 128, 128))\nout = model(x)\ny = torch.rand((2, 4, 1, 128, 128))\nloss = F.mse_loss(y, out)\nloss.backward()\n```\n\n## Citation\n```\n@article{ravuris2021skillful,\n author={Suman Ravuri and Karel Lenc and Matthew Willson and Dmitry Kangin and Remi Lam and Piotr Mirowski and Megan Fitzsimons and Maria Athanassiadou and Sheleem Kashem and Sam Madge and Rachel Prudden Amol Mandhane and Aidan Clark and Andrew Brock and Karen Simonyan and Raia Hadsell and Niall Robinson Ellen Clancy and Alberto Arribas\u2020 and Shakir Mohamed},\n title={Skillful Precipitation Nowcasting using Deep Generative Models of Radar},\n journal={Nature},\n volume={597},\n pages={672--677},\n year={2021}\n}\n```\n\n## Contributors \u2728\n\nThanks goes to these wonderful people ([emoji key](https://allcontributors.org/docs/en/emoji-key)):\n\n<!-- ALL-CONTRIBUTORS-LIST:START - Do not remove or modify this section -->\n<!-- prettier-ignore-start -->\n<!-- markdownlint-disable -->\n<table>\n <tbody>\n <tr>\n <td align=\"center\" valign=\"top\" width=\"14.28%\"><a href=\"https://www.jacobbieker.com\"><img src=\"https://avatars.githubusercontent.com/u/7170359?v=4?s=100\" width=\"100px;\" alt=\"Jacob Bieker\"/><br /><sub><b>Jacob Bieker</b></sub></a><br /><a href=\"https://github.com/openclimatefix/skillful_nowcasting/commits?author=jacobbieker\" title=\"Code\">\ud83d\udcbb</a></td>\n <td align=\"center\" valign=\"top\" width=\"14.28%\"><a href=\"http://johmathe.name/\"><img src=\"https://avatars.githubusercontent.com/u/467643?v=4?s=100\" width=\"100px;\" alt=\"Johan Mathe\"/><br /><sub><b>Johan Mathe</b></sub></a><br /><a href=\"https://github.com/openclimatefix/skillful_nowcasting/commits?author=johmathe\" title=\"Code\">\ud83d\udcbb</a></td>\n <td align=\"center\" valign=\"top\" width=\"14.28%\"><a href=\"https://github.com/ZHANGZ1YUE\"><img src=\"https://avatars.githubusercontent.com/u/93907996?v=4?s=100\" width=\"100px;\" alt=\"Z1YUE\"/><br /><sub><b>Z1YUE</b></sub></a><br /><a href=\"https://github.com/openclimatefix/skillful_nowcasting/issues?q=author%3AZHANGZ1YUE\" title=\"Bug reports\">\ud83d\udc1b</a></td>\n <td align=\"center\" valign=\"top\" width=\"14.28%\"><a href=\"https://github.com/GreenLimeSia\"><img src=\"https://avatars.githubusercontent.com/u/28706611?v=4?s=100\" width=\"100px;\" alt=\"Nan.Y\"/><br /><sub><b>Nan.Y</b></sub></a><br /><a href=\"#question-GreenLimeSia\" title=\"Answering Questions\">\ud83d\udcac</a></td>\n <td align=\"center\" valign=\"top\" width=\"14.28%\"><a href=\"https://github.com/hedaobaishui\"><img src=\"https://avatars.githubusercontent.com/u/20534146?v=4?s=100\" width=\"100px;\" alt=\"Taisanai\"/><br /><sub><b>Taisanai</b></sub></a><br /><a href=\"#question-hedaobaishui\" title=\"Answering Questions\">\ud83d\udcac</a></td>\n <td align=\"center\" valign=\"top\" width=\"14.28%\"><a href=\"https://github.com/primeoc\"><img src=\"https://avatars.githubusercontent.com/u/75205487?v=4?s=100\" width=\"100px;\" alt=\"cameron\"/><br /><sub><b>cameron</b></sub></a><br /><a href=\"#question-primeoc\" title=\"Answering Questions\">\ud83d\udcac</a></td>\n <td align=\"center\" valign=\"top\" width=\"14.28%\"><a href=\"https://github.com/zhrli\"><img src=\"https://avatars.githubusercontent.com/u/11074703?v=4?s=100\" width=\"100px;\" alt=\"zhrli\"/><br /><sub><b>zhrli</b></sub></a><br /><a href=\"#question-zhrli\" title=\"Answering Questions\">\ud83d\udcac</a></td>\n </tr>\n <tr>\n <td align=\"center\" valign=\"top\" width=\"14.28%\"><a href=\"https://github.com/najeeb-kazmi\"><img src=\"https://avatars.githubusercontent.com/u/14131235?v=4?s=100\" width=\"100px;\" alt=\"Najeeb Kazmi\"/><br /><sub><b>Najeeb Kazmi</b></sub></a><br /><a href=\"#question-najeeb-kazmi\" title=\"Answering Questions\">\ud83d\udcac</a></td>\n <td align=\"center\" valign=\"top\" width=\"14.28%\"><a href=\"https://github.com/TQRTQ\"><img src=\"https://avatars.githubusercontent.com/u/29155385?v=4?s=100\" width=\"100px;\" alt=\"TQRTQ\"/><br /><sub><b>TQRTQ</b></sub></a><br /><a href=\"#question-TQRTQ\" title=\"Answering Questions\">\ud83d\udcac</a></td>\n <td align=\"center\" valign=\"top\" width=\"14.28%\"><a href=\"https://www.linkedin.com/in/viktor-bordiuzha-93b078211\"><img src=\"https://avatars.githubusercontent.com/u/43813476?v=4?s=100\" width=\"100px;\" alt=\"Viktor Bordiuzha\"/><br /><sub><b>Viktor Bordiuzha</b></sub></a><br /><a href=\"#example-victor30608\" title=\"Examples\">\ud83d\udca1</a></td>\n <td align=\"center\" valign=\"top\" width=\"14.28%\"><a href=\"https://github.com/agijsberts\"><img src=\"https://avatars.githubusercontent.com/u/1579083?v=4?s=100\" width=\"100px;\" alt=\"agijsberts\"/><br /><sub><b>agijsberts</b></sub></a><br /><a href=\"https://github.com/openclimatefix/skillful_nowcasting/commits?author=agijsberts\" title=\"Code\">\ud83d\udcbb</a></td>\n <td align=\"center\" valign=\"top\" width=\"14.28%\"><a href=\"https://github.com/Mews\"><img src=\"https://avatars.githubusercontent.com/u/60406199?v=4?s=100\" width=\"100px;\" alt=\"Mews\"/><br /><sub><b>Mews</b></sub></a><br /><a href=\"https://github.com/openclimatefix/skillful_nowcasting/commits?author=Mews\" title=\"Tests\">\u26a0\ufe0f</a></td>\n <td align=\"center\" valign=\"top\" width=\"14.28%\"><a href=\"https://github.com/rutkovskii\"><img src=\"https://avatars.githubusercontent.com/u/70371908?v=4?s=100\" width=\"100px;\" alt=\"Aleksei Rutkovskii\"/><br /><sub><b>Aleksei Rutkovskii</b></sub></a><br /><a href=\"https://github.com/openclimatefix/skillful_nowcasting/commits?author=rutkovskii\" title=\"Code\">\ud83d\udcbb</a></td>\n </tr>\n </tbody>\n</table>\n\n<!-- markdownlint-restore -->\n<!-- prettier-ignore-end -->\n\n<!-- ALL-CONTRIBUTORS-LIST:END -->\n\nThis project follows the [all-contributors](https://github.com/all-contributors/all-contributors) specification. Contributions of any kind welcome!\n",
"bugtrack_url": null,
"license": "MIT License",
"summary": "PyTorch Skillful Nowcasting GAN Implementation",
"version": "1.4.0",
"project_urls": {
"Homepage": "https://github.com/openclimatefix/skillful_nowcasting"
},
"split_keywords": [
"artificial intelligence",
" deep learning",
" transformer",
" attention mechanism",
" metnet",
" forecasting",
" remote-sensing",
" gan"
],
"urls": [
{
"comment_text": "",
"digests": {
"blake2b_256": "f9f56dc73593a924165ffd86e2560d1aea4003f650107b4f20b2538d6dc6d55a",
"md5": "07bb9c306e8d7bc31f490a80bbfeb120",
"sha256": "c84a78e81419cd3ec6dc2447b4ac737e51035c7210c3831524166bed0edbed23"
},
"downloads": -1,
"filename": "dgmr-1.4.0-py3-none-any.whl",
"has_sig": false,
"md5_digest": "07bb9c306e8d7bc31f490a80bbfeb120",
"packagetype": "bdist_wheel",
"python_version": "py3",
"requires_python": null,
"size": 23676,
"upload_time": "2024-11-24T22:50:14",
"upload_time_iso_8601": "2024-11-24T22:50:14.816046Z",
"url": "https://files.pythonhosted.org/packages/f9/f5/6dc73593a924165ffd86e2560d1aea4003f650107b4f20b2538d6dc6d55a/dgmr-1.4.0-py3-none-any.whl",
"yanked": false,
"yanked_reason": null
},
{
"comment_text": "",
"digests": {
"blake2b_256": "b2e3bdb45e568073af2a38c90aac12ddc390d241cb5ac5bb25d51bbf5abc0bcc",
"md5": "555e4a891cdd9d8104b85d911f5bde5f",
"sha256": "a099ceaef989d1ef49a9b0f2881f01891d8ee94f04fb939e960a235cb350d2a3"
},
"downloads": -1,
"filename": "dgmr-1.4.0.tar.gz",
"has_sig": false,
"md5_digest": "555e4a891cdd9d8104b85d911f5bde5f",
"packagetype": "sdist",
"python_version": "source",
"requires_python": null,
"size": 24960,
"upload_time": "2024-11-24T22:50:15",
"upload_time_iso_8601": "2024-11-24T22:50:15.847506Z",
"url": "https://files.pythonhosted.org/packages/b2/e3/bdb45e568073af2a38c90aac12ddc390d241cb5ac5bb25d51bbf5abc0bcc/dgmr-1.4.0.tar.gz",
"yanked": false,
"yanked_reason": null
}
],
"upload_time": "2024-11-24 22:50:15",
"github": true,
"gitlab": false,
"bitbucket": false,
"codeberg": false,
"github_user": "openclimatefix",
"github_project": "skillful_nowcasting",
"travis_ci": false,
"coveralls": false,
"github_actions": true,
"requirements": [
{
"name": "torch",
"specs": []
},
{
"name": "antialiased_cnns",
"specs": []
},
{
"name": "pytorch_msssim",
"specs": []
},
{
"name": "numpy",
"specs": []
},
{
"name": "torchvision",
"specs": [
[
">=",
"0.11.0"
]
]
},
{
"name": "pytorch_lightning",
"specs": []
},
{
"name": "einops",
"specs": []
},
{
"name": "huggingface_hub",
"specs": [
[
"==",
"0.21.4"
]
]
},
{
"name": "wandb",
"specs": []
},
{
"name": "datasets",
"specs": []
},
{
"name": "tensorflow",
"specs": []
}
],
"lcname": "dgmr"
}
JSON
