Merge pull request #5 from openclimatefix/jacob/nature-paper

Match Nature Paper Implementation
openclimatefix · Nov 11, 2021 · 1743e06 · 1743e06
2 parents ffceee9 + 8af6638
commit 1743e06
Show file tree

Hide file tree

Showing 19 changed files with 1,740 additions and 702 deletions.
diff --git a/import_deepmind_weights.py b/import_deepmind_weights.py
@@ -0,0 +1,11 @@
+import tensorflow as tf
+import tensorflow_hub
+import torch
+from nowcasting_gan import DGMR
+import os
+import fsspec
+
+module = tensorflow_hub.load("/home/jacob/256x256/")
+print(module)
+print(module.signatures)
+sig_model = module.signatures["default"]
diff --git a/nowcasting_gan/__init__.py b/nowcasting_gan/__init__.py
@@ -1,4 +1,4 @@
-from .nowcasting_gan import NowcastingGAN
-from .generators import NowcastingSampler, NowcastingGenerator
-from .discriminators import NowcastingSpatialDiscriminator, NowcastingTemporalDiscriminator
+from .dgmr import DGMR
+from .generators import Sampler, Generator
+from .discriminators import SpatialDiscriminator, TemporalDiscriminator, Discriminator
 from .common import LatentConditioningStack, ContextConditioningStack