ibm-granite
/

granite-geospatial-wxc-downscaling

Model card Files Files and versions

victor-nasc commited on 7 days ago

Commit

c952bdd

·

1 Parent(s): 9c44f31

update UNET config and weights

Files changed (2) hide show

ECCC/configs/config_UNET.yaml +1 -1
ECCC/weights/best_rmse_UNET.pt +2 -2

ECCC/configs/config_UNET.yaml CHANGED Viewed

@@ -31,7 +31,7 @@ model:
   target_sigma: ./granite-geospatial-wxc-downscaling/ECCC/scalers/target_sigma.pt
   target_static_mu: ./granite-geospatial-wxc-downscaling/ECCC/scalers/target_static_mu.pt
   target_static_sigma: ./granite-geospatial-wxc-downscaling/ECCC/scalers/target_static_sigma.pt
-  embed_dim: 1024
   n_blocks_encoder: 8
   mlp_multiplier: 4
   n_heads: 16

   target_sigma: ./granite-geospatial-wxc-downscaling/ECCC/scalers/target_sigma.pt
   target_static_mu: ./granite-geospatial-wxc-downscaling/ECCC/scalers/target_static_mu.pt
   target_static_sigma: ./granite-geospatial-wxc-downscaling/ECCC/scalers/target_static_sigma.pt
+  embed_dim: 2560
   n_blocks_encoder: 8
   mlp_multiplier: 4
   n_heads: 16

ECCC/weights/best_rmse_UNET.pt CHANGED Viewed

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:fe1ebd3000d14f13d3b0d2bfacd685bbd428a14ddac8a1d408ef224ea1ea9d2c
-size 3055979467

 version https://git-lfs.github.com/spec/v1
+oid sha256:e15c6a55f8c04edd55f3441fc2da49013d8dfce4b63f7c0952206c268e01b02d
+size 17384534215