yzt15806542928 commited on
Commit
e5e1eb2
·
verified ·
1 Parent(s): 19ca0af

新增config.json文件

Browse files
Files changed (2) hide show
  1. README.md +2 -2
  2. config.json +42 -0
README.md CHANGED
@@ -54,8 +54,8 @@ This model package invokes NVIDIA's official `torch-harmonics` linear SFNO imple
54
  ### Download the Model Package
55
 
56
  ```bash
57
- hf download OneScience-Group/SFNO --local-dir ./SFNO
58
- cd SFNO
59
  ```
60
 
61
  ### Set Up the Runtime Environment
 
54
  ### Download the Model Package
55
 
56
  ```bash
57
+ hf download OneScience-Group/Spherical-Fourier-Neural-Operators --local-dir ./Spherical-Fourier-Neural-Operators
58
+ cd Spherical-Fourier-Neural-Operators
59
  ```
60
 
61
  ### Set Up the Runtime Environment
config.json ADDED
@@ -0,0 +1,42 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_name": "Spherical Fourier Neural Operator",
3
+ "model_type": "sfno",
4
+ "architectures": [
5
+ "OfficialSFNOAdapter"
6
+ ],
7
+ "framework": "PyTorch with torch-harmonics",
8
+ "domain": "spherical-dynamics",
9
+ "task": "autoregressive-spherical-field-prediction",
10
+ "implementation": {
11
+ "entry_point": "model/sfno_adapter.py",
12
+ "scope": "operator-level smoke configuration using the official torch-harmonics SFNO class"
13
+ },
14
+ "architecture": {
15
+ "family": "spherical Fourier neural operator",
16
+ "input_grid_shape": [
17
+ 17,
18
+ 32
19
+ ],
20
+ "input_grid": "equiangular",
21
+ "internal_grid": "legendre-gauss",
22
+ "input_channels": 2,
23
+ "output_channels": 2,
24
+ "embedding_size": 8,
25
+ "operator_layers": 2,
26
+ "spectral_scale_factor": 2,
27
+ "normalization": "none",
28
+ "residual_prediction": false,
29
+ "positional_embedding": "none"
30
+ },
31
+ "workflow": {
32
+ "synthetic_time_steps": 6,
33
+ "autoregressive_rollout_steps": 3,
34
+ "physical_variable_semantics": null
35
+ },
36
+ "configuration_sources": [
37
+ "model/default_config.json",
38
+ "model/config.py",
39
+ "model/sfno_adapter.py",
40
+ "README.md"
41
+ ]
42
+ }