This table displays the number of parameters across model layers for the short-wave surrogate model, as well as the output shapes of each layer. n denotes the number of atmospheric columns passed as input to the model; layers correspond to those displayed in Fig. 2.
Layer . | Output shape . | Number of parameters . |
---|---|---|
Main inputs | [(n, 49, 3)] | 0 |
Auxiliary inputs | [(n, 5)] | 0 |
|$\text{GRU}_{\downarrow }$| | [(n, 49, 128), (n, 128)] | 51 072 |
|$\text{Dense}_{1}$| | [(n, 128)] | 17 152 |
|$\text{GRU}_{\uparrow }$| | [(n, 50, 128)] | 99 072 |
|$\text{Dense}_{\text{out}}$| | [(n, 50, 2)] | 514 |
Total number of model parameters: | 167 810 |
Layer . | Output shape . | Number of parameters . |
---|---|---|
Main inputs | [(n, 49, 3)] | 0 |
Auxiliary inputs | [(n, 5)] | 0 |
|$\text{GRU}_{\downarrow }$| | [(n, 49, 128), (n, 128)] | 51 072 |
|$\text{Dense}_{1}$| | [(n, 128)] | 17 152 |
|$\text{GRU}_{\uparrow }$| | [(n, 50, 128)] | 99 072 |
|$\text{Dense}_{\text{out}}$| | [(n, 50, 2)] | 514 |
Total number of model parameters: | 167 810 |
This table displays the number of parameters across model layers for the short-wave surrogate model, as well as the output shapes of each layer. n denotes the number of atmospheric columns passed as input to the model; layers correspond to those displayed in Fig. 2.
Layer . | Output shape . | Number of parameters . |
---|---|---|
Main inputs | [(n, 49, 3)] | 0 |
Auxiliary inputs | [(n, 5)] | 0 |
|$\text{GRU}_{\downarrow }$| | [(n, 49, 128), (n, 128)] | 51 072 |
|$\text{Dense}_{1}$| | [(n, 128)] | 17 152 |
|$\text{GRU}_{\uparrow }$| | [(n, 50, 128)] | 99 072 |
|$\text{Dense}_{\text{out}}$| | [(n, 50, 2)] | 514 |
Total number of model parameters: | 167 810 |
Layer . | Output shape . | Number of parameters . |
---|---|---|
Main inputs | [(n, 49, 3)] | 0 |
Auxiliary inputs | [(n, 5)] | 0 |
|$\text{GRU}_{\downarrow }$| | [(n, 49, 128), (n, 128)] | 51 072 |
|$\text{Dense}_{1}$| | [(n, 128)] | 17 152 |
|$\text{GRU}_{\uparrow }$| | [(n, 50, 128)] | 99 072 |
|$\text{Dense}_{\text{out}}$| | [(n, 50, 2)] | 514 |
Total number of model parameters: | 167 810 |
This PDF is available to Subscribers Only
View Article Abstract & Purchase OptionsFor full access to this pdf, sign in to an existing account, or purchase an annual subscription.