yzt15806542928 commited on
Commit
ca0e050
·
verified ·
1 Parent(s): ab98816

新增config.json文件

Browse files
Files changed (1) hide show
  1. config.json +62 -0
config.json ADDED
@@ -0,0 +1,62 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_name": "AtmoRep",
3
+ "model_type": "atmorep",
4
+ "architectures": [
5
+ "TinyAtmoRep"
6
+ ],
7
+ "framework": "PyTorch",
8
+ "domain": "atmosphere",
9
+ "task": "masked-atmospheric-representation-learning",
10
+ "implementation": {
11
+ "entry_point": "model/tiny_atmorep.py",
12
+ "scope": "single-field compact validation model; official vorticity resources are retained separately"
13
+ },
14
+ "architecture": {
15
+ "family": "masked-token transformer",
16
+ "input_shape": [
17
+ 4,
18
+ 1,
19
+ 8,
20
+ 8
21
+ ],
22
+ "input_shape_order": [
23
+ "time",
24
+ "variable",
25
+ "latitude",
26
+ "longitude"
27
+ ],
28
+ "patch_shape": [
29
+ 1,
30
+ 4,
31
+ 4
32
+ ],
33
+ "embedding_size": 32,
34
+ "encoder_layers": 2,
35
+ "attention_heads": 4,
36
+ "supported_fields": 1
37
+ },
38
+ "official_resource": {
39
+ "field": "vorticity",
40
+ "model_levels": [
41
+ 96,
42
+ 105,
43
+ 114,
44
+ 123,
45
+ 137
46
+ ],
47
+ "global_grid_shape": [
48
+ 721,
49
+ 1440
50
+ ],
51
+ "spatial_resolution_degrees": 0.25,
52
+ "encoder_layers": 10,
53
+ "encoder_attention_heads": 16,
54
+ "decoder_layers": 10,
55
+ "decoder_attention_heads": 16
56
+ },
57
+ "configuration_sources": [
58
+ "model/tiny_atmorep.py",
59
+ "resources/id4nvwbetz/model_id4nvwbetz.json",
60
+ "README.md"
61
+ ]
62
+ }