lschmidt commited on
Commit
76c7d8e
·
verified ·
1 Parent(s): cc7e351

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -14
README.md CHANGED
@@ -17,17 +17,9 @@ This model is a modified version of [RCAN](https://arxiv.org/abs/1807.02758), or
17
  ## 🧠 Model Description
18
 
19
  - Based on the original RCAN architecture from `super-image`.
20
- - `sub_mean` and `add_mean` normalization layers have been **removed**, as normalization of physical variables differs from image RGB workflows.
21
  - Supports **multi-channel inputs**, currently set up for **2-channel wind fields**.
22
- - Custom head and tail layers allow flexible channel dimensions and upscaling factors (e.g., ×2, ×4).
23
- - Optional partial weight loading from pretrained RCAN (trained on DIV2K) to initialize the main body.
24
 
25
- ## 🛰️ Intended Use
26
-
27
- This model is intended for:
28
- - **Super-resolution of wind fields** in climate modeling applications.
29
- - **Downscaling wind speed or wind power-related variables** from coarse-resolution reanalysis or GCM output.
30
- - Research on generalizing deep learning architectures to geoscience and Earth system science domains.
31
 
32
  ## 🧪 Example
33
 
@@ -38,11 +30,8 @@ import torch
38
 
39
  path = hf_hub_download(repo_id="lschmidt/rcan-dsc", filename="rcan_model.py")
40
 
41
- # Dynamically load it
42
  exec(open(path).read())
43
 
44
-
45
- # Now you can call the function directly
46
  model = load_rcan()
47
 
48
  # load config
@@ -53,6 +42,5 @@ state_dict_path = hf_hub_download(repo_id="lschmidt/rcan-dsc", filename="pytorch
53
  state_dict = torch.load(state_dict_path, map_location="cpu")
54
  model.load_state_dict(state_dict, strict=False)
55
 
56
- inputs = torch.randn(1, 2, 10, 10)
57
  output = model(inputs)
58
-
 
17
  ## 🧠 Model Description
18
 
19
  - Based on the original RCAN architecture from `super-image`.
20
+ - `sub_mean` and `add_mean` normalization layers have been **removed**
21
  - Supports **multi-channel inputs**, currently set up for **2-channel wind fields**.
 
 
22
 
 
 
 
 
 
 
23
 
24
  ## 🧪 Example
25
 
 
30
 
31
  path = hf_hub_download(repo_id="lschmidt/rcan-dsc", filename="rcan_model.py")
32
 
 
33
  exec(open(path).read())
34
 
 
 
35
  model = load_rcan()
36
 
37
  # load config
 
42
  state_dict = torch.load(state_dict_path, map_location="cpu")
43
  model.load_state_dict(state_dict, strict=False)
44
 
45
+ inputs = torch.randn(1, 2, 10, 10)
46
  output = model(inputs)