romeokienzler commited on
Commit
69bd0c0
·
verified ·
1 Parent(s): 9c44f31

updating weights (#9)

Browse files

- update UNET config and weights (c952bdda89d1163f9668470f2c72a67b7cff787f)
- fix git lfs (204900263ceeeba459fa8d157e06f2db82c3d335)

.gitattributes CHANGED
@@ -40,3 +40,4 @@ ECCC/data_sample/hrdps/2022072900_000.nc filter=lfs diff=lfs merge=lfs -text
40
  ECCC/data_sample/hrdps/static_hrdps.nc filter=lfs diff=lfs merge=lfs -text
41
  *.nc filter=lfs diff=lfs merge=lfs -text
42
  *.png filter=lfs diff=lfs merge=lfs -text
 
 
40
  ECCC/data_sample/hrdps/static_hrdps.nc filter=lfs diff=lfs merge=lfs -text
41
  *.nc filter=lfs diff=lfs merge=lfs -text
42
  *.png filter=lfs diff=lfs merge=lfs -text
43
+ ECCC/weights/*.pt filter=lfs diff=lfs merge=lfs -text
ECCC/configs/config_UNET.yaml CHANGED
@@ -31,7 +31,7 @@ model:
31
  target_sigma: ./granite-geospatial-wxc-downscaling/ECCC/scalers/target_sigma.pt
32
  target_static_mu: ./granite-geospatial-wxc-downscaling/ECCC/scalers/target_static_mu.pt
33
  target_static_sigma: ./granite-geospatial-wxc-downscaling/ECCC/scalers/target_static_sigma.pt
34
- embed_dim: 1024
35
  n_blocks_encoder: 8
36
  mlp_multiplier: 4
37
  n_heads: 16
 
31
  target_sigma: ./granite-geospatial-wxc-downscaling/ECCC/scalers/target_sigma.pt
32
  target_static_mu: ./granite-geospatial-wxc-downscaling/ECCC/scalers/target_static_mu.pt
33
  target_static_sigma: ./granite-geospatial-wxc-downscaling/ECCC/scalers/target_static_sigma.pt
34
+ embed_dim: 2560
35
  n_blocks_encoder: 8
36
  mlp_multiplier: 4
37
  n_heads: 16
ECCC/weights/best_rmse_UNET.pt CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:fe1ebd3000d14f13d3b0d2bfacd685bbd428a14ddac8a1d408ef224ea1ea9d2c
3
- size 3055979467
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e15c6a55f8c04edd55f3441fc2da49013d8dfce4b63f7c0952206c268e01b02d
3
+ size 17384534215