peterdudfield commited on
Commit
5772335
·
1 Parent(s): 4de86d4

add model_config.yaml

Browse files
Files changed (1) hide show
  1. model_config.yaml +67 -0
model_config.yaml ADDED
@@ -0,0 +1,67 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ _target_: pvnet.models.multimodal.multimodal.Model
2
+ output_quantiles:
3
+ - 0.02
4
+ - 0.1
5
+ - 0.25
6
+ - 0.5
7
+ - 0.75
8
+ - 0.9
9
+ - 0.98
10
+ nwp_encoders_dict:
11
+ ecmwf:
12
+ _target_: pvnet.models.multimodal.encoders.encoders3d.DefaultPVNet
13
+ _partial_: true
14
+ in_channels: 14
15
+ out_features: 64
16
+ number_of_conv3d_layers: 4
17
+ conv3d_channels: 32
18
+ image_size_pixels: 36
19
+ sat_encoder:
20
+ _target_: pvnet.models.multimodal.encoders.encoders3d.DefaultPVNet
21
+ _partial_: true
22
+ in_channels: 11
23
+ out_features: 256
24
+ number_of_conv3d_layers: 6
25
+ conv3d_channels: 32
26
+ image_size_pixels: 144
27
+ add_image_embedding_channel: false
28
+ pv_encoder:
29
+ _target_: pvnet.models.multimodal.site_encoders.encoders.SingleAttentionNetwork
30
+ _partial_: true
31
+ num_sites: 1
32
+ out_features: 64
33
+ num_heads: 4
34
+ kdim: 64
35
+ id_embed_dim: 64
36
+ target_key: site
37
+ output_network:
38
+ _target_: pvnet.models.multimodal.linear_networks.networks.ResFCNet2
39
+ _partial_: true
40
+ fc_hidden_features: 128
41
+ n_res_blocks: 6
42
+ res_block_layers: 2
43
+ dropout_frac: 0.0
44
+ embedding_dim: 16
45
+ include_sun: true
46
+ include_gsp_yield_history: false
47
+ forecast_minutes: 2880
48
+ history_minutes: 2880
49
+ interval_minutes: 15
50
+ min_sat_delay_minutes: 60
51
+ sat_history_minutes: 90
52
+ pv_history_minutes: 2880
53
+ pv_interval_minutes: 15
54
+ nwp_history_minutes:
55
+ ecmwf: 120
56
+ nwp_forecast_minutes:
57
+ ecmwf: 2880
58
+ nwp_interval_minutes:
59
+ ecmwf: 60
60
+ optimizer:
61
+ _target_: pvnet.optimizers.EmbAdamWReduceLROnPlateau
62
+ lr: 0.0001
63
+ weight_decay: 0.01
64
+ amsgrad: true
65
+ patience: 4
66
+ factor: 0.1
67
+ threshold: 0.002