| |
| |
| ACTIVATION = @gelu |
| BASE_SHAPE_MULTIPLE = None |
| CORRECTOR_MODULE = @CustomCoordsCorrector |
| CORRECTOR_SCALE = 0.01 |
| DATA_FILTER_ATTENUATION = 0.0 |
| DYCORE_FILTER_ORDER = 3 |
| DYCORE_GRID = @GridWithWavenumbers() |
| DYCORE_INTEGRATOR = @imex_rk_sil3 |
| DYCORE_TAU = '120 minutes' |
| GLOBAL_OUT_SCALE = 0.01 |
| LATENT_SIZE = 384 |
| LAYER_SIZE = 384 |
| N_CNN_FEATURES = 32 |
| N_INNER_DYCORE_STEPS = 8 |
| N_SIGMA_LAYERS = 32 |
| N_TO_CLIP = 1 |
| NUM_BLOCKS = 5 |
| NUM_SUBSTEPS = 2 |
| PARAMETERIZATION_FILTER = @ml/SequentialStepFilter |
| POSITIONAL_LATENT_SIZE = 32 |
| REVERSE_EINSUM_ARG_ORDER = None |
| STABILITY_TAU = '4 minutes' |
| SURFACE_MODEL_LATENT_SIZE = 8 |
| SURFACE_MODEL_LAYER_SIZE = 8 |
| SURFACE_MODEL_OUTPUT_SIZE = 8 |
|
|
| |
| |
| decode/ColumnTower.checkpoint_tower = False |
| decode/ColumnTower.column_net_factory = @decode/MlpUniform |
| decode/ColumnTower.name = 'decode_tower' |
|
|
| |
| |
| encode/ColumnTower.checkpoint_tower = False |
| encode/ColumnTower.column_net_factory = @encode/MlpUniform |
| encode/ColumnTower.name = 'encode_tower' |
|
|
| |
| |
| process/ColumnTower.checkpoint_tower = False |
| process/ColumnTower.column_net_factory = @process/MlpUniform |
| process/ColumnTower.name = 'process_tower' |
|
|
| |
| |
| surface_model_decode/ColumnTower.checkpoint_tower = False |
| surface_model_decode/ColumnTower.column_net_factory = \ |
| @surface_model_decode/MlpUniform |
| surface_model_decode/ColumnTower.name = 'surface_model_decode_tower' |
|
|
| |
| |
| surface_model_encode/ColumnTower.checkpoint_tower = False |
| surface_model_encode/ColumnTower.column_net_factory = \ |
| @surface_model_encode/MlpUniform |
| surface_model_encode/ColumnTower.name = 'surface_model_encode_tower' |
|
|
| |
| |
| surface_model_process/ColumnTower.checkpoint_tower = False |
| surface_model_process/ColumnTower.column_net_factory = \ |
| @surface_model_process/MlpUniform |
| surface_model_process/ColumnTower.name = 'surface_model_process_tower' |
|
|
| |
| |
| advance/CombinedFeatures.feature_module_names_to_exclude = () |
| advance/CombinedFeatures.feature_modules = \ |
| (@EmbeddingSurfaceFeatures, |
| @EmbeddingVolumeFeatures, |
| @PressureFeatures, |
| @RadiationFeatures, |
| @LatitudeFeatures, |
| @advance/VelocityAndPrognostics, |
| @MemoryVelocityAndValues, |
| @with_grads/FloatDataFeatures, |
| @without_grads/FloatDataFeatures, |
| @LearnedPositionalFeatures) |
| advance/CombinedFeatures.features_to_exclude = () |
| advance/CombinedFeatures.features_transform_module = @advance/SequentialTransform |
| advance/CombinedFeatures.name = None |
|
|
| |
| |
| decoder_model/CombinedFeatures.feature_module_names_to_exclude = () |
| decoder_model/CombinedFeatures.feature_modules = \ |
| (@RadiationFeatures, |
| @LatitudeFeatures, |
| @model/VelocityAndPrognostics, |
| @with_grads/FloatDataFeatures, |
| @without_grads/FloatDataFeatures, |
| @LearnedPositionalFeatures) |
| decoder_model/CombinedFeatures.features_to_exclude = () |
| decoder_model/CombinedFeatures.features_transform_module = \ |
| @decoder_model/SequentialTransform |
| decoder_model/CombinedFeatures.name = None |
|
|
| |
| |
| embedding_model/CombinedFeatures.feature_module_names_to_exclude = () |
| embedding_model/CombinedFeatures.feature_modules = \ |
| (@embedding_model/VelocityAndPrognostics, @PressureFeatures) |
| embedding_model/CombinedFeatures.features_to_exclude = () |
| embedding_model/CombinedFeatures.features_transform_module = \ |
| @embedding_model/ShiftAndNormalize |
| embedding_model/CombinedFeatures.name = None |
|
|
| |
| |
| encoder_data/CombinedFeatures.feature_module_names_to_exclude = () |
| encoder_data/CombinedFeatures.feature_modules = \ |
| (@RadiationFeatures, |
| @LatitudeFeatures, |
| @encoder_data/VelocityAndPrognostics, |
| @with_grads/FloatDataFeatures, |
| @without_grads/FloatDataFeatures, |
| @LearnedPositionalFeatures) |
| encoder_data/CombinedFeatures.features_to_exclude = () |
| encoder_data/CombinedFeatures.features_transform_module = \ |
| @encoder_data/SequentialTransform |
| encoder_data/CombinedFeatures.name = None |
|
|
| |
| |
| sea_model/CombinedFeatures.feature_module_names_to_exclude = () |
| sea_model/CombinedFeatures.feature_modules = (@sea_model/ForcingFeatures,) |
| sea_model/CombinedFeatures.features_to_exclude = () |
| sea_model/CombinedFeatures.features_transform_module = @sea_model/ShiftAndNormalize |
| sea_model/CombinedFeatures.name = None |
|
|
| |
| |
| coordinate_system_from_dataset.spherical_harmonics_impl = \ |
| @RealSphericalHarmonicsWithZeroImag |
| coordinate_system_from_dataset.truncation = 'LINEAR' |
|
|
| |
| |
| CoordinateSystem.horizontal = @GridTL255() |
| CoordinateSystem.vertical = @SigmaCoordinatesEquidistant() |
|
|
| |
| |
| custom_corrds/CoordinateSystem.horizontal = %DYCORE_GRID |
| custom_corrds/CoordinateSystem.vertical = @SigmaCoordinatesEquidistant() |
|
|
| |
| |
| CustomCoordsCorrector.corrector_module = @DycoreWithPhysicsCorrector |
| CustomCoordsCorrector.custom_coords = @custom_corrds/CoordinateSystem() |
| CustomCoordsCorrector.name = None |
|
|
| |
| |
| data_to_xarray_with_renaming.additional_coords = None |
| data_to_xarray_with_renaming.attrs = None |
| data_to_xarray_with_renaming.renaming_dict = \ |
| {'geopotential': 'z', |
| 'latitude': 'lat', |
| 'longitude': 'lon', |
| 'temperature': 't', |
| 'u_component_of_wind': 'u', |
| 'v_component_of_wind': 'v'} |
| data_to_xarray_with_renaming.sample_ids = None |
| data_to_xarray_with_renaming.to_xarray_fn = @primitive_eq_to_xarray |
|
|
| |
| |
| divergence/DataExponentialFilter.attenuation = 14.0 |
| divergence/DataExponentialFilter.cutoff = 0.62 |
| divergence/DataExponentialFilter.name = None |
| divergence/DataExponentialFilter.order = 2 |
|
|
| |
| |
| lsp/DataExponentialFilter.attenuation = 8 |
| lsp/DataExponentialFilter.cutoff = 0.82 |
| lsp/DataExponentialFilter.name = None |
| lsp/DataExponentialFilter.order = 1 |
|
|
| |
| |
| orography/DataExponentialFilter.attenuation = %DATA_FILTER_ATTENUATION |
| orography/DataExponentialFilter.cutoff = 0 |
| orography/DataExponentialFilter.name = None |
| orography/DataExponentialFilter.order = 1 |
|
|
| |
| |
| q/DataExponentialFilter.attenuation = 17 |
| q/DataExponentialFilter.cutoff = 0.62 |
| q/DataExponentialFilter.name = None |
| q/DataExponentialFilter.order = 2 |
|
|
| |
| |
| temperature_variation/DataExponentialFilter.attenuation = 13.0 |
| temperature_variation/DataExponentialFilter.cutoff = 0.45 |
| temperature_variation/DataExponentialFilter.name = None |
| temperature_variation/DataExponentialFilter.order = 3 |
|
|
| |
| |
| vorticity/DataExponentialFilter.attenuation = 14 |
| vorticity/DataExponentialFilter.cutoff = 0.62 |
| vorticity/DataExponentialFilter.name = None |
| vorticity/DataExponentialFilter.order = 2 |
|
|
| |
| |
| DataNoFilter.name = None |
|
|
| |
| |
| DimensionalLearnedPrimitiveToWeatherbenchDecoder.correction_transform_module = \ |
| @decoder/SequentialTransform |
| DimensionalLearnedPrimitiveToWeatherbenchDecoder.inputs_to_units_mapping = \ |
| {'sim_time': 'dimensionless', |
| 't': 'kelvin', |
| 'tracers': {'specific_cloud_ice_water_content': 'dimensionless', |
| 'specific_cloud_liquid_water_content': 'dimensionless', |
| 'specific_humidity': 'dimensionless'}, |
| 'u': 'meter / second', |
| 'v': 'meter / second', |
| 'z': 'm**2 s**-2'} |
| DimensionalLearnedPrimitiveToWeatherbenchDecoder.modal_to_nodal_data_features_module = \ |
| @NullFeatures |
| DimensionalLearnedPrimitiveToWeatherbenchDecoder.modal_to_nodal_model_features_module = \ |
| @decoder_model/CombinedFeatures |
| DimensionalLearnedPrimitiveToWeatherbenchDecoder.name = None |
| DimensionalLearnedPrimitiveToWeatherbenchDecoder.nodal_mapping_module = \ |
| @NodalMapping |
| DimensionalLearnedPrimitiveToWeatherbenchDecoder.orography_module = \ |
| @LearnedOrography |
| DimensionalLearnedPrimitiveToWeatherbenchDecoder.prediction_mask = \ |
| {'sim_time': False, |
| 't': True, |
| 'tracers': {'specific_cloud_ice_water_content': True, |
| 'specific_cloud_liquid_water_content': True, |
| 'specific_humidity': True}, |
| 'u': True, |
| 'v': True, |
| 'z': True} |
| DimensionalLearnedPrimitiveToWeatherbenchDecoder.time_axis = 0 |
|
|
| |
| |
| DimensionalLearnedWeatherbenchToPrimitiveWithMemoryEncoder.correction_transform_module = \ |
| @encode/SequentialTransform |
| DimensionalLearnedWeatherbenchToPrimitiveWithMemoryEncoder.inputs_to_units_mapping = \ |
| {'sim_time': 'dimensionless', |
| 't': 'kelvin', |
| 'tracers': {'specific_cloud_ice_water_content': 'dimensionless', |
| 'specific_cloud_liquid_water_content': 'dimensionless', |
| 'specific_humidity': 'dimensionless'}, |
| 'u': 'meter / second', |
| 'v': 'meter / second', |
| 'z': 'm**2 s**-2'} |
| DimensionalLearnedWeatherbenchToPrimitiveWithMemoryEncoder.modal_to_nodal_data_features_module = \ |
| @encoder_data/CombinedFeatures |
| DimensionalLearnedWeatherbenchToPrimitiveWithMemoryEncoder.modal_to_nodal_model_features_module = \ |
| @NullFeatures |
| DimensionalLearnedWeatherbenchToPrimitiveWithMemoryEncoder.name = None |
| DimensionalLearnedWeatherbenchToPrimitiveWithMemoryEncoder.nodal_mapping_module = \ |
| @NodalMapping |
| DimensionalLearnedWeatherbenchToPrimitiveWithMemoryEncoder.orography_module = \ |
| @LearnedOrography |
| DimensionalLearnedWeatherbenchToPrimitiveWithMemoryEncoder.prediction_mask = \ |
| {'divergence': True, |
| 'log_surface_pressure': True, |
| 'sim_time': False, |
| 'temperature_variation': True, |
| 'tracers': {'specific_cloud_ice_water_content': True, |
| 'specific_cloud_liquid_water_content': True, |
| 'specific_humidity': True}, |
| 'vorticity': True} |
| DimensionalLearnedWeatherbenchToPrimitiveWithMemoryEncoder.time_axis = 0 |
| DimensionalLearnedWeatherbenchToPrimitiveWithMemoryEncoder.transform_module = \ |
| @EncoderCombinedTransform |
|
|
| |
| |
| DivCurlNeuralParameterization.filter_module = %PARAMETERIZATION_FILTER |
| DivCurlNeuralParameterization.modal_to_nodal_features_module = \ |
| @advance/CombinedFeatures |
| DivCurlNeuralParameterization.name = None |
| DivCurlNeuralParameterization.nodal_mapping_module = @NodalMapping |
| DivCurlNeuralParameterization.prediction_mask = \ |
| {'divergence': True, |
| 'log_surface_pressure': False, |
| 'sim_time': False, |
| 'temperature_variation': True, |
| 'tracers': {'specific_cloud_ice_water_content': True, |
| 'specific_cloud_liquid_water_content': True, |
| 'specific_humidity': True}, |
| 'vorticity': True} |
| DivCurlNeuralParameterization.tendency_transform_module = \ |
| @div_curl_tendency_outputs/SequentialTransform |
|
|
| |
| |
| DycoreWithPhysicsCorrector.checkpoint_explicit_terms = True |
| DycoreWithPhysicsCorrector.dycore_equation_module = \ |
| @MoistPrimitiveEquationsWithCloudMoisture |
| DycoreWithPhysicsCorrector.dycore_substeps = %N_INNER_DYCORE_STEPS |
| DycoreWithPhysicsCorrector.filter_module = @dycore/SequentialStepFilter |
| DycoreWithPhysicsCorrector.name = None |
| DycoreWithPhysicsCorrector.time_integrator = %DYCORE_INTEGRATOR |
|
|
| |
| |
| DynamicDataForcing.check_sim_time_errors = False |
| DynamicDataForcing.data_time_step = '6 hours' |
| DynamicDataForcing.dt_tolerance = '1 year' |
| DynamicDataForcing.inputs_to_units_mapping = \ |
| {'sea_ice_cover': 'dimensionless', |
| 'sea_surface_temperature': 'kelvin', |
| 'sim_time': 'dimensionless'} |
| DynamicDataForcing.name = None |
| DynamicDataForcing.time_axis = 0 |
|
|
| |
| |
| advance/EmbeddingSurfaceFeatures.embedding_module = @NodalLandSeaIceEmbedding |
| advance/EmbeddingSurfaceFeatures.feature_name = 'surface_embedding' |
| advance/EmbeddingSurfaceFeatures.name = None |
| advance/EmbeddingSurfaceFeatures.output_size = %SURFACE_MODEL_OUTPUT_SIZE |
|
|
| |
| |
| advance/EmbeddingVolumeFeatures.embedding_module = @ModalToNodalEmbedding |
| advance/EmbeddingVolumeFeatures.feature_name = 'CNN1D' |
| advance/EmbeddingVolumeFeatures.name = None |
| advance/EmbeddingVolumeFeatures.output_size = %N_CNN_FEATURES |
|
|
| |
| |
| EncoderCombinedTransform.name = None |
| EncoderCombinedTransform.transforms = \ |
| (@InputClipTransform, @EncoderFilterTransform) |
|
|
| |
| |
| EncoderFilterTransform.filter_modules = (@PerVariableDataFilter,) |
| EncoderFilterTransform.name = None |
|
|
| |
| |
| EpdTower.decode_tower_factory = @decode/ColumnTower |
| EpdTower.encode_tower_factory = @encode/ColumnTower |
| EpdTower.final_activation = None |
| EpdTower.latent_size = %LATENT_SIZE |
| EpdTower.name = None |
| EpdTower.num_process_blocks = %NUM_BLOCKS |
| EpdTower.post_encode_activation = None |
| EpdTower.pre_decode_activation = None |
| EpdTower.process_tower_factory = @process/ColumnTower |
|
|
| |
| |
| surface_model/EpdTower.decode_tower_factory = @surface_model_decode/ColumnTower |
| surface_model/EpdTower.encode_tower_factory = @surface_model_encode/ColumnTower |
| surface_model/EpdTower.final_activation = None |
| surface_model/EpdTower.latent_size = %SURFACE_MODEL_LATENT_SIZE |
| surface_model/EpdTower.name = None |
| surface_model/EpdTower.num_process_blocks = 1 |
| surface_model/EpdTower.post_encode_activation = None |
| surface_model/EpdTower.pre_decode_activation = None |
| surface_model/EpdTower.process_tower_factory = @surface_model_process/ColumnTower |
|
|
| |
| |
| dycore/ExponentialFilter.cutoff = 0 |
| dycore/ExponentialFilter.name = None |
| dycore/ExponentialFilter.order = %DYCORE_FILTER_ORDER |
| dycore/ExponentialFilter.tau = %DYCORE_TAU |
|
|
| |
| |
| stability/ExponentialFilter.cutoff = 0.4 |
| stability/ExponentialFilter.name = None |
| stability/ExponentialFilter.order = 6 |
| stability/ExponentialFilter.tau = %STABILITY_TAU |
|
|
| |
| |
| FilteredCustomOrography.filter_modules = (@orography/DataExponentialFilter,) |
| FilteredCustomOrography.name = None |
| FilteredCustomOrography.orography_data_path = None |
| FilteredCustomOrography.renaming_dict = {'latitude': 'lat', 'longitude': 'lon'} |
|
|
| |
| |
| with_grads/FloatDataFeatures.compute_gradients_module = @ToModalDiffOperators |
| with_grads/FloatDataFeatures.covariate_data_path = None |
| with_grads/FloatDataFeatures.covariate_keys = ('geopotential_at_surface',) |
| with_grads/FloatDataFeatures.name = None |
| with_grads/FloatDataFeatures.renaming_dict = \ |
| {'latitude': 'lat', 'longitude': 'lon'} |
|
|
| |
| |
| without_grads/FloatDataFeatures.covariate_data_path = None |
| without_grads/FloatDataFeatures.covariate_keys = ('land_sea_mask',) |
| without_grads/FloatDataFeatures.name = None |
| without_grads/FloatDataFeatures.renaming_dict = \ |
| {'latitude': 'lat', 'longitude': 'lon'} |
|
|
| |
| |
| sea_model/ForcingFeatures.forcing_to_include = ('sea_surface_temperature',) |
| sea_model/ForcingFeatures.name = None |
|
|
| |
| |
| gelu.approximate = True |
|
|
| |
| |
| |
|
|
| |
| |
| get_model_specs.custom_coords = @CoordinateSystem() |
| get_model_specs.model_time_step = '1 hour' |
| get_model_specs.reference_datetime_str = None |
| get_model_specs.reference_temperature = \ |
| [215.58614815, |
| 211.47405876, |
| 205.87815406, |
| 206.40755302, |
| 210.43452345, |
| 214.5683887, |
| 218.75303863, |
| 223.23145107, |
| 227.9710687, |
| 232.85381503, |
| 237.53588735, |
| 242.05068293, |
| 246.29986585, |
| 250.14294113, |
| 253.74839535, |
| 256.98024283, |
| 259.94441031, |
| 262.7041158, |
| 265.21752838, |
| 267.62333985, |
| 269.94462121, |
| 272.10056439, |
| 274.12518288, |
| 275.99833711, |
| 277.72759392, |
| 279.3292128, |
| 280.79178708, |
| 282.13507065, |
| 283.41832023, |
| 284.7682506, |
| 286.33945487, |
| 288.06707666] |
|
|
| |
| |
| get_physics_specs.construct_fn = @primitive_eq_specs_constructor |
|
|
| |
| |
| GridTL255.spherical_harmonics_impl = @RealSphericalHarmonicsWithZeroImag |
|
|
| |
| |
| DYCORE_GRID/GridWithWavenumbers.dealiasing = 'quadratic' |
| DYCORE_GRID/GridWithWavenumbers.latitude_spacing = 'gauss' |
| DYCORE_GRID/GridWithWavenumbers.longitude_offset = 0.0 |
| DYCORE_GRID/GridWithWavenumbers.longitude_wavenumbers = 254 |
| DYCORE_GRID/GridWithWavenumbers.radius = None |
| DYCORE_GRID/GridWithWavenumbers.spherical_harmonics_impl = \ |
| @RealSphericalHarmonicsWithZeroImag |
|
|
| |
| |
| advance/IdentityTransform.name = None |
|
|
| |
| |
| sea_model/IdentityTransform.name = None |
|
|
| |
| |
| |
|
|
| |
| |
| InputClipTransform.name = None |
| InputClipTransform.wavenumbers_to_clip = %N_TO_CLIP |
|
|
| |
| |
| advance/InverseLevelScale.keys_to_scale = \ |
| ['specific_humidity', |
| 'specific_humidity_del2', |
| 'specific_humidity_dlat', |
| 'specific_humidity_dlon'] |
| advance/InverseLevelScale.name = None |
| advance/InverseLevelScale.scales = \ |
| [8.822e-05, |
| 7.126e-05, |
| 0.0001047, |
| 0.0001858, |
| 0.0007601, |
| 0.002642, |
| 0.007242, |
| 0.01568, |
| 0.02907, |
| 0.04654, |
| 0.07084, |
| 0.09971, |
| 0.1355, |
| 0.1762, |
| 0.2243, |
| 0.2822, |
| 0.3459, |
| 0.4172, |
| 0.471, |
| 0.5286, |
| 0.5932, |
| 0.6682, |
| 0.7546, |
| 0.8532, |
| 0.9553, |
| 1.058, |
| 1.164, |
| 1.278, |
| 1.401, |
| 1.553, |
| 1.709, |
| 1.791] |
|
|
| |
| |
| decoder_model/InverseLevelScale.keys_to_scale = \ |
| ['specific_humidity', |
| 'specific_humidity_del2', |
| 'specific_humidity_dlat', |
| 'specific_humidity_dlon'] |
| decoder_model/InverseLevelScale.name = None |
| decoder_model/InverseLevelScale.scales = \ |
| [8.822e-05, |
| 7.126e-05, |
| 0.0001047, |
| 0.0001858, |
| 0.0007601, |
| 0.002642, |
| 0.007242, |
| 0.01568, |
| 0.02907, |
| 0.04654, |
| 0.07084, |
| 0.09971, |
| 0.1355, |
| 0.1762, |
| 0.2243, |
| 0.2822, |
| 0.3459, |
| 0.4172, |
| 0.471, |
| 0.5286, |
| 0.5932, |
| 0.6682, |
| 0.7546, |
| 0.8532, |
| 0.9553, |
| 1.058, |
| 1.164, |
| 1.278, |
| 1.401, |
| 1.553, |
| 1.709, |
| 1.791] |
|
|
| |
| |
| encoder_data/InverseLevelScale.keys_to_scale = \ |
| ['specific_humidity', |
| 'specific_humidity_del2', |
| 'specific_humidity_dlat', |
| 'specific_humidity_dlon'] |
| encoder_data/InverseLevelScale.name = None |
| encoder_data/InverseLevelScale.scales = \ |
| [3.823e-05, |
| 5.985e-05, |
| 7.103e-05, |
| 8.198e-05, |
| 8.836e-05, |
| 9.113e-05, |
| 7.745e-05, |
| 7.234e-05, |
| 8.308e-05, |
| 9.885e-05, |
| 0.0001443, |
| 0.0002999, |
| 0.001019, |
| 0.002773, |
| 0.006194, |
| 0.01195, |
| 0.02037, |
| 0.04634, |
| 0.08648, |
| 0.1406, |
| 0.2104, |
| 0.3017, |
| 0.4097, |
| 0.4949, |
| 0.5891, |
| 0.716, |
| 0.8645, |
| 0.9418, |
| 1.019, |
| 1.098, |
| 1.178, |
| 1.262, |
| 1.35, |
| 1.454, |
| 1.581, |
| 1.677, |
| 1.713] |
|
|
| |
| |
| decoder/InverseShiftAndNormalize.global_scale = %CORRECTOR_SCALE |
| decoder/InverseShiftAndNormalize.name = None |
| decoder/InverseShiftAndNormalize.scales = \ |
| {'sim_time': 0.0, |
| 't': 27.99, |
| 'tracers': {'specific_cloud_ice_water_content': 8.255e-06, |
| 'specific_cloud_liquid_water_content': 2.182e-05, |
| 'specific_humidity': 0.003493}, |
| 'u': 0.01935, |
| 'v': 0.01038, |
| 'z': 0.1496} |
| decoder/InverseShiftAndNormalize.shifts = \ |
| {'sim_time': 0.0, |
| 't': 0.0, |
| 'tracers': {'specific_cloud_ice_water_content': 0.0, |
| 'specific_cloud_liquid_water_content': 0.0, |
| 'specific_humidity': 0.0}, |
| 'u': 0.0, |
| 'v': 0.0, |
| 'z': 0.0} |
|
|
| |
| |
| div_curl_tendency_outputs/InverseShiftAndNormalize.global_scale = %GLOBAL_OUT_SCALE |
| div_curl_tendency_outputs/InverseShiftAndNormalize.name = None |
| div_curl_tendency_outputs/InverseShiftAndNormalize.scales = \ |
| {'log_surface_pressure': 0.05021, |
| 'sim_time': 0.0, |
| 'temperature_variation': 33.94, |
| 'tracers': {'specific_cloud_ice_water_content': 4.85e-05, |
| 'specific_cloud_liquid_water_content': 9.693e-05, |
| 'specific_humidity': 0.006182}, |
| 'u': 0.05863, |
| 'v': 0.0516} |
| div_curl_tendency_outputs/InverseShiftAndNormalize.shifts = \ |
| {'log_surface_pressure': 0.0, |
| 'sim_time': 0.0, |
| 'temperature_variation': 0.0, |
| 'tracers': {'specific_cloud_ice_water_content': 0.0, |
| 'specific_cloud_liquid_water_content': 0.0, |
| 'specific_humidity': 0.0}, |
| 'u': 0.0, |
| 'v': 0.0} |
|
|
| |
| |
| encoder/InverseShiftAndNormalize.global_scale = %CORRECTOR_SCALE |
| encoder/InverseShiftAndNormalize.name = None |
| encoder/InverseShiftAndNormalize.scales = \ |
| {'divergence': 0.1078, |
| 'log_surface_pressure': 0.1134, |
| 'sim_time': 0.0, |
| 'temperature_variation': 15.04, |
| 'tracers': {'specific_cloud_ice_water_content': 9.812e-06, |
| 'specific_cloud_liquid_water_content': 2.039e-05, |
| 'specific_humidity': 0.003305}, |
| 'vorticity': 0.2831} |
| encoder/InverseShiftAndNormalize.shifts = \ |
| {'divergence': 0.0, |
| 'log_surface_pressure': 0.0, |
| 'sim_time': 0.0, |
| 'temperature_variation': 0.0, |
| 'tracers': {'specific_cloud_ice_water_content': 0.0, |
| 'specific_cloud_liquid_water_content': 0.0, |
| 'specific_humidity': 0.0}, |
| 'vorticity': 0.0} |
|
|
| |
| |
| advance/LatitudeFeatures.name = None |
|
|
| |
| |
| decoder_model/LatitudeFeatures.name = None |
|
|
| |
| |
| encoder_data/LatitudeFeatures.name = None |
|
|
| |
| |
| LearnedOrography.base_orography_module = @FilteredCustomOrography |
| LearnedOrography.correction_scale = 2e-06 |
| LearnedOrography.name = None |
|
|
| |
| |
| advance/LearnedPositionalFeatures.latent_size = %POSITIONAL_LATENT_SIZE |
| advance/LearnedPositionalFeatures.name = None |
| advance/LearnedPositionalFeatures.scale = 1.0 |
|
|
| |
| |
| decoder_model/LearnedPositionalFeatures.latent_size = %POSITIONAL_LATENT_SIZE |
| decoder_model/LearnedPositionalFeatures.name = None |
| decoder_model/LearnedPositionalFeatures.scale = 1.0 |
|
|
| |
| |
| encoder_data/LearnedPositionalFeatures.latent_size = %POSITIONAL_LATENT_SIZE |
| encoder_data/LearnedPositionalFeatures.name = None |
| encoder_data/LearnedPositionalFeatures.scale = 1.0 |
|
|
| |
| |
| decoder/LevelScale.keys_to_scale = ['specific_humidity'] |
| decoder/LevelScale.name = None |
| decoder/LevelScale.scales = \ |
| [3.823e-05, |
| 5.985e-05, |
| 7.103e-05, |
| 8.198e-05, |
| 8.836e-05, |
| 9.113e-05, |
| 7.745e-05, |
| 7.234e-05, |
| 8.308e-05, |
| 9.885e-05, |
| 0.0001443, |
| 0.0002999, |
| 0.001019, |
| 0.002773, |
| 0.006194, |
| 0.01195, |
| 0.02037, |
| 0.04634, |
| 0.08648, |
| 0.1406, |
| 0.2104, |
| 0.3017, |
| 0.4097, |
| 0.4949, |
| 0.5891, |
| 0.716, |
| 0.8645, |
| 0.9418, |
| 1.019, |
| 1.098, |
| 1.178, |
| 1.262, |
| 1.35, |
| 1.454, |
| 1.581, |
| 1.677, |
| 1.713] |
|
|
| |
| |
| div_curl_tendency_outputs/LevelScale.keys_to_scale = ['specific_humidity'] |
| div_curl_tendency_outputs/LevelScale.name = None |
| div_curl_tendency_outputs/LevelScale.scales = \ |
| [0.0001303, |
| 0.0002029, |
| 0.0002682, |
| 0.0004815, |
| 0.001437, |
| 0.004719, |
| 0.01328, |
| 0.03024, |
| 0.05782, |
| 0.09671, |
| 0.1485, |
| 0.2107, |
| 0.2892, |
| 0.3723, |
| 0.4705, |
| 0.5785, |
| 0.6885, |
| 0.8155, |
| 0.9028, |
| 0.9944, |
| 1.103, |
| 1.197, |
| 1.31, |
| 1.431, |
| 1.526, |
| 1.58, |
| 1.61, |
| 1.624, |
| 1.634, |
| 1.649, |
| 1.729, |
| 1.824] |
|
|
| |
| |
| encode/LevelScale.keys_to_scale = ['specific_humidity'] |
| encode/LevelScale.name = None |
| encode/LevelScale.scales = \ |
| [8.822e-05, |
| 7.126e-05, |
| 0.0001047, |
| 0.0001858, |
| 0.0007601, |
| 0.002642, |
| 0.007242, |
| 0.01568, |
| 0.02907, |
| 0.04654, |
| 0.07084, |
| 0.09971, |
| 0.1355, |
| 0.1762, |
| 0.2243, |
| 0.2822, |
| 0.3459, |
| 0.4172, |
| 0.471, |
| 0.5286, |
| 0.5932, |
| 0.6682, |
| 0.7546, |
| 0.8532, |
| 0.9553, |
| 1.058, |
| 1.164, |
| 1.278, |
| 1.401, |
| 1.553, |
| 1.709, |
| 1.791] |
|
|
| |
| |
| advance/MemoryVelocityAndValues.fields_to_include = None |
| advance/MemoryVelocityAndValues.name = None |
|
|
| |
| |
| decode/MlpUniform.activate_final = False |
| decode/MlpUniform.activation = %ACTIVATION |
| decode/MlpUniform.b_init = None |
| decode/MlpUniform.b_init_final = None |
| decode/MlpUniform.name = None |
| decode/MlpUniform.num_hidden_layers = 0 |
| decode/MlpUniform.num_hidden_units = %LAYER_SIZE |
| decode/MlpUniform.w_init = None |
| decode/MlpUniform.w_init_final = None |
| decode/MlpUniform.with_bias = False |
|
|
| |
| |
| encode/MlpUniform.activate_final = False |
| encode/MlpUniform.activation = %ACTIVATION |
| encode/MlpUniform.b_init = None |
| encode/MlpUniform.b_init_final = None |
| encode/MlpUniform.name = None |
| encode/MlpUniform.num_hidden_layers = 0 |
| encode/MlpUniform.num_hidden_units = 0 |
| encode/MlpUniform.w_init = None |
| encode/MlpUniform.w_init_final = None |
| encode/MlpUniform.with_bias = True |
|
|
| |
| |
| process/MlpUniform.activate_final = False |
| process/MlpUniform.activation = %ACTIVATION |
| process/MlpUniform.b_init = None |
| process/MlpUniform.b_init_final = None |
| process/MlpUniform.name = None |
| process/MlpUniform.num_hidden_layers = 3 |
| process/MlpUniform.num_hidden_units = %LAYER_SIZE |
| process/MlpUniform.w_init = None |
| process/MlpUniform.w_init_final = None |
| process/MlpUniform.with_bias = True |
|
|
| |
| |
| surface_model_decode/MlpUniform.activate_final = False |
| surface_model_decode/MlpUniform.activation = %ACTIVATION |
| surface_model_decode/MlpUniform.b_init = None |
| surface_model_decode/MlpUniform.b_init_final = None |
| surface_model_decode/MlpUniform.name = None |
| surface_model_decode/MlpUniform.num_hidden_layers = 1 |
| surface_model_decode/MlpUniform.num_hidden_units = %SURFACE_MODEL_LAYER_SIZE |
| surface_model_decode/MlpUniform.w_init = None |
| surface_model_decode/MlpUniform.w_init_final = None |
| surface_model_decode/MlpUniform.with_bias = False |
|
|
| |
| |
| surface_model_encode/MlpUniform.activate_final = False |
| surface_model_encode/MlpUniform.activation = %ACTIVATION |
| surface_model_encode/MlpUniform.b_init = None |
| surface_model_encode/MlpUniform.b_init_final = None |
| surface_model_encode/MlpUniform.name = None |
| surface_model_encode/MlpUniform.num_hidden_layers = 0 |
| surface_model_encode/MlpUniform.num_hidden_units = 0 |
| surface_model_encode/MlpUniform.w_init = None |
| surface_model_encode/MlpUniform.w_init_final = None |
| surface_model_encode/MlpUniform.with_bias = True |
|
|
| |
| |
| surface_model_process/MlpUniform.activate_final = False |
| surface_model_process/MlpUniform.activation = %ACTIVATION |
| surface_model_process/MlpUniform.b_init = None |
| surface_model_process/MlpUniform.b_init_final = None |
| surface_model_process/MlpUniform.name = None |
| surface_model_process/MlpUniform.num_hidden_layers = 3 |
| surface_model_process/MlpUniform.num_hidden_units = %SURFACE_MODEL_LAYER_SIZE |
| surface_model_process/MlpUniform.w_init = None |
| surface_model_process/MlpUniform.w_init_final = None |
| surface_model_process/MlpUniform.with_bias = True |
|
|
| |
| |
| advance/ModalToNodalEmbedding.modal_to_nodal_features_module = \ |
| @embedding_model/CombinedFeatures |
| advance/ModalToNodalEmbedding.name = None |
| advance/ModalToNodalEmbedding.nodal_mapping_module = @NodalVolumeMapping |
| advance/ModalToNodalEmbedding.output_transform_module = @IdentityTransform |
|
|
| |
| |
| sea_model/ModalToNodalEmbedding.modal_to_nodal_features_module = \ |
| @sea_model/CombinedFeatures |
| sea_model/ModalToNodalEmbedding.name = None |
| sea_model/ModalToNodalEmbedding.nodal_mapping_module = @sea_model/NodalMapping |
| sea_model/ModalToNodalEmbedding.output_transform_module = @IdentityTransform |
|
|
| |
| |
| MoistPrimitiveEquationsWithCloudMoisture.include_vertical_advection = True |
| MoistPrimitiveEquationsWithCloudMoisture.name = None |
| MoistPrimitiveEquationsWithCloudMoisture.orography_module = @LearnedOrography |
|
|
| |
| |
| advance/NodalLandSeaIceEmbedding.land_embedding = None |
| advance/NodalLandSeaIceEmbedding.name = None |
| advance/NodalLandSeaIceEmbedding.sea_embedding = @sea_model/ModalToNodalEmbedding |
| advance/NodalLandSeaIceEmbedding.sea_ice_embedding = None |
| advance/NodalLandSeaIceEmbedding.static_vars_ds_path = None |
|
|
| |
| |
| NodalMapping.name = None |
| NodalMapping.tower_factory = @EpdTower |
|
|
| |
| |
| sea_model/NodalMapping.name = None |
| sea_model/NodalMapping.tower_factory = @surface_model/EpdTower |
|
|
| |
| |
| NodalVolumeMapping.name = None |
| NodalVolumeMapping.tower_factory = @VerticalConvTower |
|
|
| |
| |
| NullFeatures.name = None |
|
|
| |
| |
| PerVariableDataFilter.name = None |
| PerVariableDataFilter.per_variable_filters = \ |
| {'divergence': @divergence/DataExponentialFilter, |
| 'log_surface_pressure': @lsp/DataExponentialFilter, |
| 'sim_time': @DataNoFilter, |
| 'temperature_variation': @temperature_variation/DataExponentialFilter, |
| 'tracers': {'specific_cloud_ice_water_content': @q/DataExponentialFilter, |
| 'specific_cloud_liquid_water_content': @q/DataExponentialFilter, |
| 'specific_humidity': @q/DataExponentialFilter}, |
| 'vorticity': @vorticity/DataExponentialFilter} |
|
|
| |
| |
| advance/PressureFeatures.name = None |
|
|
| |
| |
| embedding_model/PressureFeatures.name = None |
|
|
| |
| |
| primitive_eq_specs_constructor.scale = @GET_ATMOSPHERIC_SCALE() |
|
|
| |
| |
| |
|
|
| |
| |
| |
|
|
| |
| |
| advance/RadiationFeatures.name = None |
|
|
| |
| |
| decoder_model/RadiationFeatures.name = None |
|
|
| |
| |
| encoder_data/RadiationFeatures.name = None |
|
|
| |
| |
| RealSphericalHarmonicsWithZeroImag.base_shape_multiple = %BASE_SHAPE_MULTIPLE |
| RealSphericalHarmonicsWithZeroImag.reverse_einsum_arg_order = \ |
| %REVERSE_EINSUM_ARG_ORDER |
| RealSphericalHarmonicsWithZeroImag.stacked_fourier_transforms = None |
| RealSphericalHarmonicsWithZeroImag.transform_precision = 'tensorfloat32' |
|
|
| |
| |
| orography/RealSphericalHarmonicsWithZeroImag.base_shape_multiple = \ |
| %BASE_SHAPE_MULTIPLE |
| orography/RealSphericalHarmonicsWithZeroImag.reverse_einsum_arg_order = \ |
| %REVERSE_EINSUM_ARG_ORDER |
| orography/RealSphericalHarmonicsWithZeroImag.stacked_fourier_transforms = None |
| orography/RealSphericalHarmonicsWithZeroImag.transform_precision = 'tensorfloat32' |
|
|
| |
| |
| dycore/SequentialStepFilter.filter_modules = \ |
| (@dycore/ExponentialFilter, @stability/ExponentialFilter) |
| dycore/SequentialStepFilter.name = None |
|
|
| |
| |
| ml/SequentialStepFilter.filter_modules = (@stability/ExponentialFilter,) |
| ml/SequentialStepFilter.name = None |
|
|
| |
| |
| advance/SequentialTransform.name = None |
| advance/SequentialTransform.transform_modules = \ |
| (@advance/ShiftAndNormalize, |
| @advance/InverseLevelScale, |
| @advance/TruncateSigmaLevels, |
| @SoftClip) |
|
|
| |
| |
| decoder/SequentialTransform.name = None |
| decoder/SequentialTransform.transform_modules = \ |
| (@decoder/InverseShiftAndNormalize, @decoder/LevelScale) |
|
|
| |
| |
| decoder_model/SequentialTransform.name = None |
| decoder_model/SequentialTransform.transform_modules = \ |
| (@decoder_model/ShiftAndNormalize, |
| @decoder_model/InverseLevelScale, |
| @decoder_model/TruncateSigmaLevels) |
|
|
| |
| |
| div_curl_tendency_outputs/SequentialTransform.name = None |
| div_curl_tendency_outputs/SequentialTransform.transform_modules = \ |
| (@div_curl_tendency_outputs/InverseShiftAndNormalize, |
| @div_curl_tendency_outputs/LevelScale) |
|
|
| |
| |
| encode/SequentialTransform.name = None |
| encode/SequentialTransform.transform_modules = \ |
| (@encoder/InverseShiftAndNormalize, @encode/LevelScale) |
|
|
| |
| |
| encoder_data/SequentialTransform.name = None |
| encoder_data/SequentialTransform.transform_modules = \ |
| (@encoder_data/ShiftAndNormalize, @encoder_data/InverseLevelScale) |
|
|
| |
| |
| advance/ShiftAndNormalize.features_to_exclude = () |
| advance/ShiftAndNormalize.global_scale = None |
| advance/ShiftAndNormalize.name = None |
| advance/ShiftAndNormalize.scales = \ |
| {'CNN1D_0': 0.1, |
| 'CNN1D_1': 0.1, |
| 'CNN1D_10': 0.1, |
| 'CNN1D_11': 0.1, |
| 'CNN1D_12': 0.1, |
| 'CNN1D_13': 0.1, |
| 'CNN1D_14': 0.1, |
| 'CNN1D_15': 0.1, |
| 'CNN1D_16': 0.1, |
| 'CNN1D_17': 0.1, |
| 'CNN1D_18': 0.1, |
| 'CNN1D_19': 0.1, |
| 'CNN1D_2': 0.1, |
| 'CNN1D_20': 0.1, |
| 'CNN1D_21': 0.1, |
| 'CNN1D_22': 0.1, |
| 'CNN1D_23': 0.1, |
| 'CNN1D_24': 0.1, |
| 'CNN1D_25': 0.1, |
| 'CNN1D_26': 0.1, |
| 'CNN1D_27': 0.1, |
| 'CNN1D_28': 0.1, |
| 'CNN1D_29': 0.1, |
| 'CNN1D_3': 0.1, |
| 'CNN1D_30': 0.1, |
| 'CNN1D_31': 0.1, |
| 'CNN1D_32': 0.1, |
| 'CNN1D_33': 0.1, |
| 'CNN1D_34': 0.1, |
| 'CNN1D_35': 0.1, |
| 'CNN1D_36': 0.1, |
| 'CNN1D_37': 0.1, |
| 'CNN1D_38': 0.1, |
| 'CNN1D_39': 0.1, |
| 'CNN1D_4': 0.1, |
| 'CNN1D_40': 0.1, |
| 'CNN1D_41': 0.1, |
| 'CNN1D_42': 0.1, |
| 'CNN1D_43': 0.1, |
| 'CNN1D_44': 0.1, |
| 'CNN1D_45': 0.1, |
| 'CNN1D_46': 0.1, |
| 'CNN1D_47': 0.1, |
| 'CNN1D_48': 0.1, |
| 'CNN1D_49': 0.1, |
| 'CNN1D_5': 0.1, |
| 'CNN1D_50': 0.1, |
| 'CNN1D_51': 0.1, |
| 'CNN1D_52': 0.1, |
| 'CNN1D_53': 0.1, |
| 'CNN1D_54': 0.1, |
| 'CNN1D_55': 0.1, |
| 'CNN1D_56': 0.1, |
| 'CNN1D_57': 0.1, |
| 'CNN1D_58': 0.1, |
| 'CNN1D_59': 0.1, |
| 'CNN1D_6': 0.1, |
| 'CNN1D_60': 0.1, |
| 'CNN1D_61': 0.1, |
| 'CNN1D_62': 0.1, |
| 'CNN1D_63': 0.1, |
| 'CNN1D_7': 0.1, |
| 'CNN1D_8': 0.1, |
| 'CNN1D_9': 0.1, |
| 'cos_latitude': 0.3068, |
| 'divergence': 0.1078, |
| 'divergence_del2': 1825.0, |
| 'divergence_dlat': 8.5, |
| 'divergence_dlon': 8.842, |
| 'geopotential_at_surface': 0.009579, |
| 'geopotential_at_surface_del2': 28.66, |
| 'geopotential_at_surface_dlat': 0.1882, |
| 'geopotential_at_surface_dlon': 0.1564, |
| 'land_sea_mask': 0.456, |
| 'learned_positional_features': 1.0, |
| 'log_surface_pressure': 0.1134, |
| 'log_surface_pressure_del2': 264.6, |
| 'log_surface_pressure_dlat': 1.656, |
| 'log_surface_pressure_dlon': 1.645, |
| 'memory_divergence': 0.1078, |
| 'memory_log_surface_pressure': 0.1134, |
| 'memory_specific_cloud_ice_water_content': 9.812e-06, |
| 'memory_specific_cloud_liquid_water_content': 2.039e-05, |
| 'memory_specific_humidity': 0.003305, |
| 'memory_temperature_variation': 15.04, |
| 'memory_u': 0.01489, |
| 'memory_v': 0.01022, |
| 'memory_vorticity': 0.2831, |
| 'pressure': 1.644, |
| 'radiation': 0.2862, |
| 'sea_surface_temperature': 11.93, |
| 'sin_latitude': 0.7064, |
| 'specific_cloud_ice_water_content': 9.812e-06, |
| 'specific_cloud_ice_water_content_del2': 0.08441, |
| 'specific_cloud_ice_water_content_dlat': 0.0004684, |
| 'specific_cloud_ice_water_content_dlon': 0.0004646, |
| 'specific_cloud_liquid_water_content': 2.039e-05, |
| 'specific_cloud_liquid_water_content_del2': 0.1802, |
| 'specific_cloud_liquid_water_content_dlat': 0.0009877, |
| 'specific_cloud_liquid_water_content_dlon': 0.0009565, |
| 'specific_humidity': 0.003305, |
| 'specific_humidity_del2': 4.442, |
| 'specific_humidity_dlat': 0.0319, |
| 'specific_humidity_dlon': 0.02678, |
| 'surface_embedding': 1.0, |
| 'temperature_variation': 15.04, |
| 'temperature_variation_del2': 10460.0, |
| 'temperature_variation_dlat': 82.59, |
| 'temperature_variation_dlon': 77.07, |
| 'u': 0.01489, |
| 'u_del2': 18.46, |
| 'u_dlat': 0.2197, |
| 'u_dlon': 0.1677, |
| 'v': 0.01022, |
| 'v_del2': 17.2, |
| 'v_dlat': 0.169, |
| 'v_dlon': 0.2083, |
| 'vorticity': 0.2831, |
| 'vorticity_del2': 2831.0, |
| 'vorticity_dlat': 14.96, |
| 'vorticity_dlon': 14.91} |
| advance/ShiftAndNormalize.shifts = \ |
| {'CNN1D_0': 0.0, |
| 'CNN1D_1': 0.0, |
| 'CNN1D_10': 0.0, |
| 'CNN1D_11': 0.0, |
| 'CNN1D_12': 0.0, |
| 'CNN1D_13': 0.0, |
| 'CNN1D_14': 0.0, |
| 'CNN1D_15': 0.0, |
| 'CNN1D_16': 0.0, |
| 'CNN1D_17': 0.0, |
| 'CNN1D_18': 0.0, |
| 'CNN1D_19': 0.0, |
| 'CNN1D_2': 0.0, |
| 'CNN1D_20': 0.0, |
| 'CNN1D_21': 0.0, |
| 'CNN1D_22': 0.0, |
| 'CNN1D_23': 0.0, |
| 'CNN1D_24': 0.0, |
| 'CNN1D_25': 0.0, |
| 'CNN1D_26': 0.0, |
| 'CNN1D_27': 0.0, |
| 'CNN1D_28': 0.0, |
| 'CNN1D_29': 0.0, |
| 'CNN1D_3': 0.0, |
| 'CNN1D_30': 0.0, |
| 'CNN1D_31': 0.0, |
| 'CNN1D_32': 0.0, |
| 'CNN1D_33': 0.0, |
| 'CNN1D_34': 0.0, |
| 'CNN1D_35': 0.0, |
| 'CNN1D_36': 0.0, |
| 'CNN1D_37': 0.0, |
| 'CNN1D_38': 0.0, |
| 'CNN1D_39': 0.0, |
| 'CNN1D_4': 0.0, |
| 'CNN1D_40': 0.0, |
| 'CNN1D_41': 0.0, |
| 'CNN1D_42': 0.0, |
| 'CNN1D_43': 0.0, |
| 'CNN1D_44': 0.0, |
| 'CNN1D_45': 0.0, |
| 'CNN1D_46': 0.0, |
| 'CNN1D_47': 0.0, |
| 'CNN1D_48': 0.0, |
| 'CNN1D_49': 0.0, |
| 'CNN1D_5': 0.0, |
| 'CNN1D_50': 0.0, |
| 'CNN1D_51': 0.0, |
| 'CNN1D_52': 0.0, |
| 'CNN1D_53': 0.0, |
| 'CNN1D_54': 0.0, |
| 'CNN1D_55': 0.0, |
| 'CNN1D_56': 0.0, |
| 'CNN1D_57': 0.0, |
| 'CNN1D_58': 0.0, |
| 'CNN1D_59': 0.0, |
| 'CNN1D_6': 0.0, |
| 'CNN1D_60': 0.0, |
| 'CNN1D_61': 0.0, |
| 'CNN1D_62': 0.0, |
| 'CNN1D_63': 0.0, |
| 'CNN1D_7': 0.0, |
| 'CNN1D_8': 0.0, |
| 'CNN1D_9': 0.0, |
| 'cos_latitude': 0.638, |
| 'divergence': -0.0, |
| 'divergence_del2': -0.015, |
| 'divergence_dlat': -0.001, |
| 'divergence_dlon': 0.0, |
| 'geopotential_at_surface': 0.004, |
| 'geopotential_at_surface_del2': -0.048, |
| 'geopotential_at_surface_dlat': -0.01, |
| 'geopotential_at_surface_dlon': -0.0, |
| 'land_sea_mask': 0.335, |
| 'learned_positional_features': 0.0, |
| 'log_surface_pressure': 1.715, |
| 'log_surface_pressure_del2': 0.59, |
| 'log_surface_pressure_dlat': 0.126, |
| 'log_surface_pressure_dlon': 0.0, |
| 'memory_divergence': -0.0, |
| 'memory_log_surface_pressure': 1.715, |
| 'memory_specific_cloud_ice_water_content': 0.0, |
| 'memory_specific_cloud_liquid_water_content': 0.0, |
| 'memory_specific_humidity': 0.0, |
| 'memory_temperature_variation': -4.98, |
| 'memory_u': 0.007, |
| 'memory_v': -0.0, |
| 'memory_vorticity': -0.002, |
| 'pressure': 2.796, |
| 'radiation': 0.213, |
| 'sea_surface_temperature': 285.14, |
| 'sin_latitude': -0.0, |
| 'specific_cloud_ice_water_content': 0.0, |
| 'specific_cloud_ice_water_content_del2': 0.0, |
| 'specific_cloud_ice_water_content_dlat': 0.0, |
| 'specific_cloud_ice_water_content_dlon': -0.0, |
| 'specific_cloud_liquid_water_content': 0.0, |
| 'specific_cloud_liquid_water_content_del2': -0.0, |
| 'specific_cloud_liquid_water_content_dlat': 0.0, |
| 'specific_cloud_liquid_water_content_dlon': -0.0, |
| 'specific_humidity': 0.0, |
| 'specific_humidity_del2': 0.0, |
| 'specific_humidity_dlat': 0.0, |
| 'specific_humidity_dlon': 0.0, |
| 'surface_embedding': 0.0, |
| 'temperature_variation': -4.98, |
| 'temperature_variation_del2': 43.608, |
| 'temperature_variation_dlat': 4.843, |
| 'temperature_variation_dlon': 0.0, |
| 'u': 0.007, |
| 'u_del2': 0.242, |
| 'u_dlat': 0.002, |
| 'u_dlon': 0.0, |
| 'v': -0.0, |
| 'v_del2': -0.0, |
| 'v_dlat': -0.0, |
| 'v_dlon': 0.0, |
| 'vorticity': -0.002, |
| 'vorticity_del2': -0.365, |
| 'vorticity_dlat': 0.041, |
| 'vorticity_dlon': 0.0} |
|
|
| |
| |
| decoder_model/ShiftAndNormalize.features_to_exclude = () |
| decoder_model/ShiftAndNormalize.global_scale = None |
| decoder_model/ShiftAndNormalize.name = None |
| decoder_model/ShiftAndNormalize.scales = \ |
| {'CNN1D_0': 0.1, |
| 'CNN1D_1': 0.1, |
| 'CNN1D_10': 0.1, |
| 'CNN1D_11': 0.1, |
| 'CNN1D_12': 0.1, |
| 'CNN1D_13': 0.1, |
| 'CNN1D_14': 0.1, |
| 'CNN1D_15': 0.1, |
| 'CNN1D_16': 0.1, |
| 'CNN1D_17': 0.1, |
| 'CNN1D_18': 0.1, |
| 'CNN1D_19': 0.1, |
| 'CNN1D_2': 0.1, |
| 'CNN1D_20': 0.1, |
| 'CNN1D_21': 0.1, |
| 'CNN1D_22': 0.1, |
| 'CNN1D_23': 0.1, |
| 'CNN1D_24': 0.1, |
| 'CNN1D_25': 0.1, |
| 'CNN1D_26': 0.1, |
| 'CNN1D_27': 0.1, |
| 'CNN1D_28': 0.1, |
| 'CNN1D_29': 0.1, |
| 'CNN1D_3': 0.1, |
| 'CNN1D_30': 0.1, |
| 'CNN1D_31': 0.1, |
| 'CNN1D_32': 0.1, |
| 'CNN1D_33': 0.1, |
| 'CNN1D_34': 0.1, |
| 'CNN1D_35': 0.1, |
| 'CNN1D_36': 0.1, |
| 'CNN1D_37': 0.1, |
| 'CNN1D_38': 0.1, |
| 'CNN1D_39': 0.1, |
| 'CNN1D_4': 0.1, |
| 'CNN1D_40': 0.1, |
| 'CNN1D_41': 0.1, |
| 'CNN1D_42': 0.1, |
| 'CNN1D_43': 0.1, |
| 'CNN1D_44': 0.1, |
| 'CNN1D_45': 0.1, |
| 'CNN1D_46': 0.1, |
| 'CNN1D_47': 0.1, |
| 'CNN1D_48': 0.1, |
| 'CNN1D_49': 0.1, |
| 'CNN1D_5': 0.1, |
| 'CNN1D_50': 0.1, |
| 'CNN1D_51': 0.1, |
| 'CNN1D_52': 0.1, |
| 'CNN1D_53': 0.1, |
| 'CNN1D_54': 0.1, |
| 'CNN1D_55': 0.1, |
| 'CNN1D_56': 0.1, |
| 'CNN1D_57': 0.1, |
| 'CNN1D_58': 0.1, |
| 'CNN1D_59': 0.1, |
| 'CNN1D_6': 0.1, |
| 'CNN1D_60': 0.1, |
| 'CNN1D_61': 0.1, |
| 'CNN1D_62': 0.1, |
| 'CNN1D_63': 0.1, |
| 'CNN1D_7': 0.1, |
| 'CNN1D_8': 0.1, |
| 'CNN1D_9': 0.1, |
| 'cos_latitude': 0.3068, |
| 'divergence': 0.1078, |
| 'divergence_del2': 1825.0, |
| 'divergence_dlat': 8.5, |
| 'divergence_dlon': 8.842, |
| 'geopotential_at_surface': 0.009579, |
| 'geopotential_at_surface_del2': 28.66, |
| 'geopotential_at_surface_dlat': 0.1882, |
| 'geopotential_at_surface_dlon': 0.1564, |
| 'land_sea_mask': 0.456, |
| 'learned_positional_features': 1.0, |
| 'log_surface_pressure': 0.1134, |
| 'log_surface_pressure_del2': 264.6, |
| 'log_surface_pressure_dlat': 1.656, |
| 'log_surface_pressure_dlon': 1.645, |
| 'memory_divergence': 0.1078, |
| 'memory_log_surface_pressure': 0.1134, |
| 'memory_specific_cloud_ice_water_content': 9.812e-06, |
| 'memory_specific_cloud_liquid_water_content': 2.039e-05, |
| 'memory_specific_humidity': 0.003305, |
| 'memory_temperature_variation': 15.04, |
| 'memory_u': 0.01489, |
| 'memory_v': 0.01022, |
| 'memory_vorticity': 0.2831, |
| 'pressure': 1.644, |
| 'radiation': 0.2862, |
| 'sea_surface_temperature': 11.93, |
| 'sin_latitude': 0.7064, |
| 'specific_cloud_ice_water_content': 9.812e-06, |
| 'specific_cloud_ice_water_content_del2': 0.08441, |
| 'specific_cloud_ice_water_content_dlat': 0.0004684, |
| 'specific_cloud_ice_water_content_dlon': 0.0004646, |
| 'specific_cloud_liquid_water_content': 2.039e-05, |
| 'specific_cloud_liquid_water_content_del2': 0.1802, |
| 'specific_cloud_liquid_water_content_dlat': 0.0009877, |
| 'specific_cloud_liquid_water_content_dlon': 0.0009565, |
| 'specific_humidity': 0.003305, |
| 'specific_humidity_del2': 4.442, |
| 'specific_humidity_dlat': 0.0319, |
| 'specific_humidity_dlon': 0.02678, |
| 'surface_embedding': 1.0, |
| 'temperature_variation': 15.04, |
| 'temperature_variation_del2': 10460.0, |
| 'temperature_variation_dlat': 82.59, |
| 'temperature_variation_dlon': 77.07, |
| 'u': 0.01489, |
| 'u_del2': 18.46, |
| 'u_dlat': 0.2197, |
| 'u_dlon': 0.1677, |
| 'v': 0.01022, |
| 'v_del2': 17.2, |
| 'v_dlat': 0.169, |
| 'v_dlon': 0.2083, |
| 'vorticity': 0.2831, |
| 'vorticity_del2': 2831.0, |
| 'vorticity_dlat': 14.96, |
| 'vorticity_dlon': 14.91} |
| decoder_model/ShiftAndNormalize.shifts = \ |
| {'CNN1D_0': 0.0, |
| 'CNN1D_1': 0.0, |
| 'CNN1D_10': 0.0, |
| 'CNN1D_11': 0.0, |
| 'CNN1D_12': 0.0, |
| 'CNN1D_13': 0.0, |
| 'CNN1D_14': 0.0, |
| 'CNN1D_15': 0.0, |
| 'CNN1D_16': 0.0, |
| 'CNN1D_17': 0.0, |
| 'CNN1D_18': 0.0, |
| 'CNN1D_19': 0.0, |
| 'CNN1D_2': 0.0, |
| 'CNN1D_20': 0.0, |
| 'CNN1D_21': 0.0, |
| 'CNN1D_22': 0.0, |
| 'CNN1D_23': 0.0, |
| 'CNN1D_24': 0.0, |
| 'CNN1D_25': 0.0, |
| 'CNN1D_26': 0.0, |
| 'CNN1D_27': 0.0, |
| 'CNN1D_28': 0.0, |
| 'CNN1D_29': 0.0, |
| 'CNN1D_3': 0.0, |
| 'CNN1D_30': 0.0, |
| 'CNN1D_31': 0.0, |
| 'CNN1D_32': 0.0, |
| 'CNN1D_33': 0.0, |
| 'CNN1D_34': 0.0, |
| 'CNN1D_35': 0.0, |
| 'CNN1D_36': 0.0, |
| 'CNN1D_37': 0.0, |
| 'CNN1D_38': 0.0, |
| 'CNN1D_39': 0.0, |
| 'CNN1D_4': 0.0, |
| 'CNN1D_40': 0.0, |
| 'CNN1D_41': 0.0, |
| 'CNN1D_42': 0.0, |
| 'CNN1D_43': 0.0, |
| 'CNN1D_44': 0.0, |
| 'CNN1D_45': 0.0, |
| 'CNN1D_46': 0.0, |
| 'CNN1D_47': 0.0, |
| 'CNN1D_48': 0.0, |
| 'CNN1D_49': 0.0, |
| 'CNN1D_5': 0.0, |
| 'CNN1D_50': 0.0, |
| 'CNN1D_51': 0.0, |
| 'CNN1D_52': 0.0, |
| 'CNN1D_53': 0.0, |
| 'CNN1D_54': 0.0, |
| 'CNN1D_55': 0.0, |
| 'CNN1D_56': 0.0, |
| 'CNN1D_57': 0.0, |
| 'CNN1D_58': 0.0, |
| 'CNN1D_59': 0.0, |
| 'CNN1D_6': 0.0, |
| 'CNN1D_60': 0.0, |
| 'CNN1D_61': 0.0, |
| 'CNN1D_62': 0.0, |
| 'CNN1D_63': 0.0, |
| 'CNN1D_7': 0.0, |
| 'CNN1D_8': 0.0, |
| 'CNN1D_9': 0.0, |
| 'cos_latitude': 0.638, |
| 'divergence': -0.0, |
| 'divergence_del2': -0.015, |
| 'divergence_dlat': -0.001, |
| 'divergence_dlon': 0.0, |
| 'geopotential_at_surface': 0.004, |
| 'geopotential_at_surface_del2': -0.048, |
| 'geopotential_at_surface_dlat': -0.01, |
| 'geopotential_at_surface_dlon': -0.0, |
| 'land_sea_mask': 0.335, |
| 'learned_positional_features': 0.0, |
| 'log_surface_pressure': 1.715, |
| 'log_surface_pressure_del2': 0.59, |
| 'log_surface_pressure_dlat': 0.126, |
| 'log_surface_pressure_dlon': 0.0, |
| 'memory_divergence': -0.0, |
| 'memory_log_surface_pressure': 1.715, |
| 'memory_specific_cloud_ice_water_content': 0.0, |
| 'memory_specific_cloud_liquid_water_content': 0.0, |
| 'memory_specific_humidity': 0.0, |
| 'memory_temperature_variation': -4.98, |
| 'memory_u': 0.007, |
| 'memory_v': -0.0, |
| 'memory_vorticity': -0.002, |
| 'pressure': 2.796, |
| 'radiation': 0.213, |
| 'sea_surface_temperature': 285.14, |
| 'sin_latitude': -0.0, |
| 'specific_cloud_ice_water_content': 0.0, |
| 'specific_cloud_ice_water_content_del2': 0.0, |
| 'specific_cloud_ice_water_content_dlat': 0.0, |
| 'specific_cloud_ice_water_content_dlon': -0.0, |
| 'specific_cloud_liquid_water_content': 0.0, |
| 'specific_cloud_liquid_water_content_del2': -0.0, |
| 'specific_cloud_liquid_water_content_dlat': 0.0, |
| 'specific_cloud_liquid_water_content_dlon': -0.0, |
| 'specific_humidity': 0.0, |
| 'specific_humidity_del2': 0.0, |
| 'specific_humidity_dlat': 0.0, |
| 'specific_humidity_dlon': 0.0, |
| 'surface_embedding': 0.0, |
| 'temperature_variation': -4.98, |
| 'temperature_variation_del2': 43.608, |
| 'temperature_variation_dlat': 4.843, |
| 'temperature_variation_dlon': 0.0, |
| 'u': 0.007, |
| 'u_del2': 0.242, |
| 'u_dlat': 0.002, |
| 'u_dlon': 0.0, |
| 'v': -0.0, |
| 'v_del2': -0.0, |
| 'v_dlat': -0.0, |
| 'v_dlon': 0.0, |
| 'vorticity': -0.002, |
| 'vorticity_del2': -0.365, |
| 'vorticity_dlat': 0.041, |
| 'vorticity_dlon': 0.0} |
|
|
| |
| |
| embedding_model/ShiftAndNormalize.features_to_exclude = () |
| embedding_model/ShiftAndNormalize.global_scale = None |
| embedding_model/ShiftAndNormalize.name = None |
| embedding_model/ShiftAndNormalize.scales = \ |
| {'CNN1D_0': 0.1, |
| 'CNN1D_1': 0.1, |
| 'CNN1D_10': 0.1, |
| 'CNN1D_11': 0.1, |
| 'CNN1D_12': 0.1, |
| 'CNN1D_13': 0.1, |
| 'CNN1D_14': 0.1, |
| 'CNN1D_15': 0.1, |
| 'CNN1D_16': 0.1, |
| 'CNN1D_17': 0.1, |
| 'CNN1D_18': 0.1, |
| 'CNN1D_19': 0.1, |
| 'CNN1D_2': 0.1, |
| 'CNN1D_20': 0.1, |
| 'CNN1D_21': 0.1, |
| 'CNN1D_22': 0.1, |
| 'CNN1D_23': 0.1, |
| 'CNN1D_24': 0.1, |
| 'CNN1D_25': 0.1, |
| 'CNN1D_26': 0.1, |
| 'CNN1D_27': 0.1, |
| 'CNN1D_28': 0.1, |
| 'CNN1D_29': 0.1, |
| 'CNN1D_3': 0.1, |
| 'CNN1D_30': 0.1, |
| 'CNN1D_31': 0.1, |
| 'CNN1D_32': 0.1, |
| 'CNN1D_33': 0.1, |
| 'CNN1D_34': 0.1, |
| 'CNN1D_35': 0.1, |
| 'CNN1D_36': 0.1, |
| 'CNN1D_37': 0.1, |
| 'CNN1D_38': 0.1, |
| 'CNN1D_39': 0.1, |
| 'CNN1D_4': 0.1, |
| 'CNN1D_40': 0.1, |
| 'CNN1D_41': 0.1, |
| 'CNN1D_42': 0.1, |
| 'CNN1D_43': 0.1, |
| 'CNN1D_44': 0.1, |
| 'CNN1D_45': 0.1, |
| 'CNN1D_46': 0.1, |
| 'CNN1D_47': 0.1, |
| 'CNN1D_48': 0.1, |
| 'CNN1D_49': 0.1, |
| 'CNN1D_5': 0.1, |
| 'CNN1D_50': 0.1, |
| 'CNN1D_51': 0.1, |
| 'CNN1D_52': 0.1, |
| 'CNN1D_53': 0.1, |
| 'CNN1D_54': 0.1, |
| 'CNN1D_55': 0.1, |
| 'CNN1D_56': 0.1, |
| 'CNN1D_57': 0.1, |
| 'CNN1D_58': 0.1, |
| 'CNN1D_59': 0.1, |
| 'CNN1D_6': 0.1, |
| 'CNN1D_60': 0.1, |
| 'CNN1D_61': 0.1, |
| 'CNN1D_62': 0.1, |
| 'CNN1D_63': 0.1, |
| 'CNN1D_7': 0.1, |
| 'CNN1D_8': 0.1, |
| 'CNN1D_9': 0.1, |
| 'cos_latitude': 0.3068, |
| 'divergence': 0.1078, |
| 'divergence_del2': 1825.0, |
| 'divergence_dlat': 8.5, |
| 'divergence_dlon': 8.842, |
| 'geopotential_at_surface': 0.009579, |
| 'geopotential_at_surface_del2': 28.66, |
| 'geopotential_at_surface_dlat': 0.1882, |
| 'geopotential_at_surface_dlon': 0.1564, |
| 'land_sea_mask': 0.456, |
| 'learned_positional_features': 1.0, |
| 'log_surface_pressure': 0.1134, |
| 'log_surface_pressure_del2': 264.6, |
| 'log_surface_pressure_dlat': 1.656, |
| 'log_surface_pressure_dlon': 1.645, |
| 'memory_divergence': 0.1078, |
| 'memory_log_surface_pressure': 0.1134, |
| 'memory_specific_cloud_ice_water_content': 9.812e-06, |
| 'memory_specific_cloud_liquid_water_content': 2.039e-05, |
| 'memory_specific_humidity': 0.003305, |
| 'memory_temperature_variation': 15.04, |
| 'memory_u': 0.01489, |
| 'memory_v': 0.01022, |
| 'memory_vorticity': 0.2831, |
| 'pressure': 1.644, |
| 'radiation': 0.2862, |
| 'sea_surface_temperature': 11.93, |
| 'sin_latitude': 0.7064, |
| 'specific_cloud_ice_water_content': 9.812e-06, |
| 'specific_cloud_ice_water_content_del2': 0.08441, |
| 'specific_cloud_ice_water_content_dlat': 0.0004684, |
| 'specific_cloud_ice_water_content_dlon': 0.0004646, |
| 'specific_cloud_liquid_water_content': 2.039e-05, |
| 'specific_cloud_liquid_water_content_del2': 0.1802, |
| 'specific_cloud_liquid_water_content_dlat': 0.0009877, |
| 'specific_cloud_liquid_water_content_dlon': 0.0009565, |
| 'specific_humidity': 0.003305, |
| 'specific_humidity_del2': 4.442, |
| 'specific_humidity_dlat': 0.0319, |
| 'specific_humidity_dlon': 0.02678, |
| 'surface_embedding': 1.0, |
| 'temperature_variation': 15.04, |
| 'temperature_variation_del2': 10460.0, |
| 'temperature_variation_dlat': 82.59, |
| 'temperature_variation_dlon': 77.07, |
| 'u': 0.01489, |
| 'u_del2': 18.46, |
| 'u_dlat': 0.2197, |
| 'u_dlon': 0.1677, |
| 'v': 0.01022, |
| 'v_del2': 17.2, |
| 'v_dlat': 0.169, |
| 'v_dlon': 0.2083, |
| 'vorticity': 0.2831, |
| 'vorticity_del2': 2831.0, |
| 'vorticity_dlat': 14.96, |
| 'vorticity_dlon': 14.91} |
| embedding_model/ShiftAndNormalize.shifts = \ |
| {'CNN1D_0': 0.0, |
| 'CNN1D_1': 0.0, |
| 'CNN1D_10': 0.0, |
| 'CNN1D_11': 0.0, |
| 'CNN1D_12': 0.0, |
| 'CNN1D_13': 0.0, |
| 'CNN1D_14': 0.0, |
| 'CNN1D_15': 0.0, |
| 'CNN1D_16': 0.0, |
| 'CNN1D_17': 0.0, |
| 'CNN1D_18': 0.0, |
| 'CNN1D_19': 0.0, |
| 'CNN1D_2': 0.0, |
| 'CNN1D_20': 0.0, |
| 'CNN1D_21': 0.0, |
| 'CNN1D_22': 0.0, |
| 'CNN1D_23': 0.0, |
| 'CNN1D_24': 0.0, |
| 'CNN1D_25': 0.0, |
| 'CNN1D_26': 0.0, |
| 'CNN1D_27': 0.0, |
| 'CNN1D_28': 0.0, |
| 'CNN1D_29': 0.0, |
| 'CNN1D_3': 0.0, |
| 'CNN1D_30': 0.0, |
| 'CNN1D_31': 0.0, |
| 'CNN1D_32': 0.0, |
| 'CNN1D_33': 0.0, |
| 'CNN1D_34': 0.0, |
| 'CNN1D_35': 0.0, |
| 'CNN1D_36': 0.0, |
| 'CNN1D_37': 0.0, |
| 'CNN1D_38': 0.0, |
| 'CNN1D_39': 0.0, |
| 'CNN1D_4': 0.0, |
| 'CNN1D_40': 0.0, |
| 'CNN1D_41': 0.0, |
| 'CNN1D_42': 0.0, |
| 'CNN1D_43': 0.0, |
| 'CNN1D_44': 0.0, |
| 'CNN1D_45': 0.0, |
| 'CNN1D_46': 0.0, |
| 'CNN1D_47': 0.0, |
| 'CNN1D_48': 0.0, |
| 'CNN1D_49': 0.0, |
| 'CNN1D_5': 0.0, |
| 'CNN1D_50': 0.0, |
| 'CNN1D_51': 0.0, |
| 'CNN1D_52': 0.0, |
| 'CNN1D_53': 0.0, |
| 'CNN1D_54': 0.0, |
| 'CNN1D_55': 0.0, |
| 'CNN1D_56': 0.0, |
| 'CNN1D_57': 0.0, |
| 'CNN1D_58': 0.0, |
| 'CNN1D_59': 0.0, |
| 'CNN1D_6': 0.0, |
| 'CNN1D_60': 0.0, |
| 'CNN1D_61': 0.0, |
| 'CNN1D_62': 0.0, |
| 'CNN1D_63': 0.0, |
| 'CNN1D_7': 0.0, |
| 'CNN1D_8': 0.0, |
| 'CNN1D_9': 0.0, |
| 'cos_latitude': 0.638, |
| 'divergence': -0.0, |
| 'divergence_del2': -0.015, |
| 'divergence_dlat': -0.001, |
| 'divergence_dlon': 0.0, |
| 'geopotential_at_surface': 0.004, |
| 'geopotential_at_surface_del2': -0.048, |
| 'geopotential_at_surface_dlat': -0.01, |
| 'geopotential_at_surface_dlon': -0.0, |
| 'land_sea_mask': 0.335, |
| 'learned_positional_features': 0.0, |
| 'log_surface_pressure': 1.715, |
| 'log_surface_pressure_del2': 0.59, |
| 'log_surface_pressure_dlat': 0.126, |
| 'log_surface_pressure_dlon': 0.0, |
| 'memory_divergence': -0.0, |
| 'memory_log_surface_pressure': 1.715, |
| 'memory_specific_cloud_ice_water_content': 0.0, |
| 'memory_specific_cloud_liquid_water_content': 0.0, |
| 'memory_specific_humidity': 0.0, |
| 'memory_temperature_variation': -4.98, |
| 'memory_u': 0.007, |
| 'memory_v': -0.0, |
| 'memory_vorticity': -0.002, |
| 'pressure': 2.796, |
| 'radiation': 0.213, |
| 'sea_surface_temperature': 285.14, |
| 'sin_latitude': -0.0, |
| 'specific_cloud_ice_water_content': 0.0, |
| 'specific_cloud_ice_water_content_del2': 0.0, |
| 'specific_cloud_ice_water_content_dlat': 0.0, |
| 'specific_cloud_ice_water_content_dlon': -0.0, |
| 'specific_cloud_liquid_water_content': 0.0, |
| 'specific_cloud_liquid_water_content_del2': -0.0, |
| 'specific_cloud_liquid_water_content_dlat': 0.0, |
| 'specific_cloud_liquid_water_content_dlon': -0.0, |
| 'specific_humidity': 0.0, |
| 'specific_humidity_del2': 0.0, |
| 'specific_humidity_dlat': 0.0, |
| 'specific_humidity_dlon': 0.0, |
| 'surface_embedding': 0.0, |
| 'temperature_variation': -4.98, |
| 'temperature_variation_del2': 43.608, |
| 'temperature_variation_dlat': 4.843, |
| 'temperature_variation_dlon': 0.0, |
| 'u': 0.007, |
| 'u_del2': 0.242, |
| 'u_dlat': 0.002, |
| 'u_dlon': 0.0, |
| 'v': -0.0, |
| 'v_del2': -0.0, |
| 'v_dlat': -0.0, |
| 'v_dlon': 0.0, |
| 'vorticity': -0.002, |
| 'vorticity_del2': -0.365, |
| 'vorticity_dlat': 0.041, |
| 'vorticity_dlon': 0.0} |
|
|
| |
| |
| encoder_data/ShiftAndNormalize.features_to_exclude = () |
| encoder_data/ShiftAndNormalize.global_scale = None |
| encoder_data/ShiftAndNormalize.name = None |
| encoder_data/ShiftAndNormalize.scales = \ |
| {'cos_latitude': 0.3068, |
| 'geopotential_at_surface': 0.009579, |
| 'geopotential_at_surface_del2': 28.66, |
| 'geopotential_at_surface_dlat': 0.1882, |
| 'geopotential_at_surface_dlon': 0.1564, |
| 'land_sea_mask': 0.456, |
| 'learned_positional_features': 1.0, |
| 'radiation': 0.2862, |
| 'sin_latitude': 0.7064, |
| 'specific_cloud_ice_water_content': 8.255e-06, |
| 'specific_cloud_ice_water_content_del2': 0.09943, |
| 'specific_cloud_ice_water_content_dlat': 0.0005741, |
| 'specific_cloud_ice_water_content_dlon': 0.0004452, |
| 'specific_cloud_liquid_water_content': 2.182e-05, |
| 'specific_cloud_liquid_water_content_del2': 0.2881, |
| 'specific_cloud_liquid_water_content_dlat': 0.001674, |
| 'specific_cloud_liquid_water_content_dlon': 0.001224, |
| 'specific_humidity': 0.003493, |
| 'specific_humidity_del2': 5.716, |
| 'specific_humidity_dlat': 0.0396, |
| 'specific_humidity_dlon': 0.02857, |
| 't': 27.99, |
| 't_del2': 9788.0, |
| 't_dlat': 71.14, |
| 't_dlon': 62.14, |
| 'u': 0.01935, |
| 'u_del2': 27.4, |
| 'u_dlat': 0.1876, |
| 'u_dlon': 0.1712, |
| 'v': 0.01038, |
| 'v_del2': 27.11, |
| 'v_dlat': 0.1326, |
| 'v_dlon': 0.1998, |
| 'z': 0.1496, |
| 'z_del2': 0.6108, |
| 'z_dlat': 0.01606, |
| 'z_dlon': 0.009139} |
| encoder_data/ShiftAndNormalize.shifts = \ |
| {'cos_latitude': 0.638, |
| 'geopotential_at_surface': 0.004, |
| 'geopotential_at_surface_del2': -0.048, |
| 'geopotential_at_surface_dlat': -0.01, |
| 'geopotential_at_surface_dlon': -0.0, |
| 'land_sea_mask': 0.335, |
| 'learned_positional_features': 0.0, |
| 'radiation': 0.213, |
| 'sin_latitude': -0.0, |
| 'specific_cloud_ice_water_content': 0.0, |
| 'specific_cloud_ice_water_content_del2': -0.0, |
| 'specific_cloud_ice_water_content_dlat': 0.0, |
| 'specific_cloud_ice_water_content_dlon': 0.0, |
| 'specific_cloud_liquid_water_content': 0.0, |
| 'specific_cloud_liquid_water_content_del2': -0.0, |
| 'specific_cloud_liquid_water_content_dlat': 0.0, |
| 'specific_cloud_liquid_water_content_dlon': 0.0, |
| 'specific_humidity': 0.0, |
| 'specific_humidity_del2': 0.0, |
| 'specific_humidity_dlat': 0.0, |
| 'specific_humidity_dlon': 0.0, |
| 't': 247.045, |
| 't_del2': 37.572, |
| 't_dlat': 1.755, |
| 't_dlon': -0.0, |
| 'u': 0.007, |
| 'u_del2': 0.056, |
| 'u_dlat': 0.0, |
| 'u_dlon': 0.0, |
| 'v': 0.0, |
| 'v_del2': 0.008, |
| 'v_dlat': 0.0, |
| 'v_dlon': -0.0, |
| 'z': 0.145, |
| 'z_del2': 0.011, |
| 'z_dlat': 0.001, |
| 'z_dlon': 0.0} |
|
|
| |
| |
| sea_model/ShiftAndNormalize.features_to_exclude = () |
| sea_model/ShiftAndNormalize.global_scale = None |
| sea_model/ShiftAndNormalize.name = None |
| sea_model/ShiftAndNormalize.scales = \ |
| {'CNN1D_0': 0.1, |
| 'CNN1D_1': 0.1, |
| 'CNN1D_10': 0.1, |
| 'CNN1D_11': 0.1, |
| 'CNN1D_12': 0.1, |
| 'CNN1D_13': 0.1, |
| 'CNN1D_14': 0.1, |
| 'CNN1D_15': 0.1, |
| 'CNN1D_16': 0.1, |
| 'CNN1D_17': 0.1, |
| 'CNN1D_18': 0.1, |
| 'CNN1D_19': 0.1, |
| 'CNN1D_2': 0.1, |
| 'CNN1D_20': 0.1, |
| 'CNN1D_21': 0.1, |
| 'CNN1D_22': 0.1, |
| 'CNN1D_23': 0.1, |
| 'CNN1D_24': 0.1, |
| 'CNN1D_25': 0.1, |
| 'CNN1D_26': 0.1, |
| 'CNN1D_27': 0.1, |
| 'CNN1D_28': 0.1, |
| 'CNN1D_29': 0.1, |
| 'CNN1D_3': 0.1, |
| 'CNN1D_30': 0.1, |
| 'CNN1D_31': 0.1, |
| 'CNN1D_32': 0.1, |
| 'CNN1D_33': 0.1, |
| 'CNN1D_34': 0.1, |
| 'CNN1D_35': 0.1, |
| 'CNN1D_36': 0.1, |
| 'CNN1D_37': 0.1, |
| 'CNN1D_38': 0.1, |
| 'CNN1D_39': 0.1, |
| 'CNN1D_4': 0.1, |
| 'CNN1D_40': 0.1, |
| 'CNN1D_41': 0.1, |
| 'CNN1D_42': 0.1, |
| 'CNN1D_43': 0.1, |
| 'CNN1D_44': 0.1, |
| 'CNN1D_45': 0.1, |
| 'CNN1D_46': 0.1, |
| 'CNN1D_47': 0.1, |
| 'CNN1D_48': 0.1, |
| 'CNN1D_49': 0.1, |
| 'CNN1D_5': 0.1, |
| 'CNN1D_50': 0.1, |
| 'CNN1D_51': 0.1, |
| 'CNN1D_52': 0.1, |
| 'CNN1D_53': 0.1, |
| 'CNN1D_54': 0.1, |
| 'CNN1D_55': 0.1, |
| 'CNN1D_56': 0.1, |
| 'CNN1D_57': 0.1, |
| 'CNN1D_58': 0.1, |
| 'CNN1D_59': 0.1, |
| 'CNN1D_6': 0.1, |
| 'CNN1D_60': 0.1, |
| 'CNN1D_61': 0.1, |
| 'CNN1D_62': 0.1, |
| 'CNN1D_63': 0.1, |
| 'CNN1D_7': 0.1, |
| 'CNN1D_8': 0.1, |
| 'CNN1D_9': 0.1, |
| 'cos_latitude': 0.3068, |
| 'divergence': 0.1078, |
| 'divergence_del2': 1825.0, |
| 'divergence_dlat': 8.5, |
| 'divergence_dlon': 8.842, |
| 'geopotential_at_surface': 0.009579, |
| 'geopotential_at_surface_del2': 28.66, |
| 'geopotential_at_surface_dlat': 0.1882, |
| 'geopotential_at_surface_dlon': 0.1564, |
| 'land_sea_mask': 0.456, |
| 'learned_positional_features': 1.0, |
| 'log_surface_pressure': 0.1134, |
| 'log_surface_pressure_del2': 264.6, |
| 'log_surface_pressure_dlat': 1.656, |
| 'log_surface_pressure_dlon': 1.645, |
| 'memory_divergence': 0.1078, |
| 'memory_log_surface_pressure': 0.1134, |
| 'memory_specific_cloud_ice_water_content': 9.812e-06, |
| 'memory_specific_cloud_liquid_water_content': 2.039e-05, |
| 'memory_specific_humidity': 0.003305, |
| 'memory_temperature_variation': 15.04, |
| 'memory_u': 0.01489, |
| 'memory_v': 0.01022, |
| 'memory_vorticity': 0.2831, |
| 'pressure': 1.644, |
| 'radiation': 0.2862, |
| 'sea_surface_temperature': 11.93, |
| 'sin_latitude': 0.7064, |
| 'specific_cloud_ice_water_content': 9.812e-06, |
| 'specific_cloud_ice_water_content_del2': 0.08441, |
| 'specific_cloud_ice_water_content_dlat': 0.0004684, |
| 'specific_cloud_ice_water_content_dlon': 0.0004646, |
| 'specific_cloud_liquid_water_content': 2.039e-05, |
| 'specific_cloud_liquid_water_content_del2': 0.1802, |
| 'specific_cloud_liquid_water_content_dlat': 0.0009877, |
| 'specific_cloud_liquid_water_content_dlon': 0.0009565, |
| 'specific_humidity': 0.003305, |
| 'specific_humidity_del2': 4.442, |
| 'specific_humidity_dlat': 0.0319, |
| 'specific_humidity_dlon': 0.02678, |
| 'surface_embedding': 1.0, |
| 'temperature_variation': 15.04, |
| 'temperature_variation_del2': 10460.0, |
| 'temperature_variation_dlat': 82.59, |
| 'temperature_variation_dlon': 77.07, |
| 'u': 0.01489, |
| 'u_del2': 18.46, |
| 'u_dlat': 0.2197, |
| 'u_dlon': 0.1677, |
| 'v': 0.01022, |
| 'v_del2': 17.2, |
| 'v_dlat': 0.169, |
| 'v_dlon': 0.2083, |
| 'vorticity': 0.2831, |
| 'vorticity_del2': 2831.0, |
| 'vorticity_dlat': 14.96, |
| 'vorticity_dlon': 14.91} |
| sea_model/ShiftAndNormalize.shifts = \ |
| {'CNN1D_0': 0.0, |
| 'CNN1D_1': 0.0, |
| 'CNN1D_10': 0.0, |
| 'CNN1D_11': 0.0, |
| 'CNN1D_12': 0.0, |
| 'CNN1D_13': 0.0, |
| 'CNN1D_14': 0.0, |
| 'CNN1D_15': 0.0, |
| 'CNN1D_16': 0.0, |
| 'CNN1D_17': 0.0, |
| 'CNN1D_18': 0.0, |
| 'CNN1D_19': 0.0, |
| 'CNN1D_2': 0.0, |
| 'CNN1D_20': 0.0, |
| 'CNN1D_21': 0.0, |
| 'CNN1D_22': 0.0, |
| 'CNN1D_23': 0.0, |
| 'CNN1D_24': 0.0, |
| 'CNN1D_25': 0.0, |
| 'CNN1D_26': 0.0, |
| 'CNN1D_27': 0.0, |
| 'CNN1D_28': 0.0, |
| 'CNN1D_29': 0.0, |
| 'CNN1D_3': 0.0, |
| 'CNN1D_30': 0.0, |
| 'CNN1D_31': 0.0, |
| 'CNN1D_32': 0.0, |
| 'CNN1D_33': 0.0, |
| 'CNN1D_34': 0.0, |
| 'CNN1D_35': 0.0, |
| 'CNN1D_36': 0.0, |
| 'CNN1D_37': 0.0, |
| 'CNN1D_38': 0.0, |
| 'CNN1D_39': 0.0, |
| 'CNN1D_4': 0.0, |
| 'CNN1D_40': 0.0, |
| 'CNN1D_41': 0.0, |
| 'CNN1D_42': 0.0, |
| 'CNN1D_43': 0.0, |
| 'CNN1D_44': 0.0, |
| 'CNN1D_45': 0.0, |
| 'CNN1D_46': 0.0, |
| 'CNN1D_47': 0.0, |
| 'CNN1D_48': 0.0, |
| 'CNN1D_49': 0.0, |
| 'CNN1D_5': 0.0, |
| 'CNN1D_50': 0.0, |
| 'CNN1D_51': 0.0, |
| 'CNN1D_52': 0.0, |
| 'CNN1D_53': 0.0, |
| 'CNN1D_54': 0.0, |
| 'CNN1D_55': 0.0, |
| 'CNN1D_56': 0.0, |
| 'CNN1D_57': 0.0, |
| 'CNN1D_58': 0.0, |
| 'CNN1D_59': 0.0, |
| 'CNN1D_6': 0.0, |
| 'CNN1D_60': 0.0, |
| 'CNN1D_61': 0.0, |
| 'CNN1D_62': 0.0, |
| 'CNN1D_63': 0.0, |
| 'CNN1D_7': 0.0, |
| 'CNN1D_8': 0.0, |
| 'CNN1D_9': 0.0, |
| 'cos_latitude': 0.638, |
| 'divergence': -0.0, |
| 'divergence_del2': -0.015, |
| 'divergence_dlat': -0.001, |
| 'divergence_dlon': 0.0, |
| 'geopotential_at_surface': 0.004, |
| 'geopotential_at_surface_del2': -0.048, |
| 'geopotential_at_surface_dlat': -0.01, |
| 'geopotential_at_surface_dlon': -0.0, |
| 'land_sea_mask': 0.335, |
| 'learned_positional_features': 0.0, |
| 'log_surface_pressure': 1.715, |
| 'log_surface_pressure_del2': 0.59, |
| 'log_surface_pressure_dlat': 0.126, |
| 'log_surface_pressure_dlon': 0.0, |
| 'memory_divergence': -0.0, |
| 'memory_log_surface_pressure': 1.715, |
| 'memory_specific_cloud_ice_water_content': 0.0, |
| 'memory_specific_cloud_liquid_water_content': 0.0, |
| 'memory_specific_humidity': 0.0, |
| 'memory_temperature_variation': -4.98, |
| 'memory_u': 0.007, |
| 'memory_v': -0.0, |
| 'memory_vorticity': -0.002, |
| 'pressure': 2.796, |
| 'radiation': 0.213, |
| 'sea_surface_temperature': 285.14, |
| 'sin_latitude': -0.0, |
| 'specific_cloud_ice_water_content': 0.0, |
| 'specific_cloud_ice_water_content_del2': 0.0, |
| 'specific_cloud_ice_water_content_dlat': 0.0, |
| 'specific_cloud_ice_water_content_dlon': -0.0, |
| 'specific_cloud_liquid_water_content': 0.0, |
| 'specific_cloud_liquid_water_content_del2': -0.0, |
| 'specific_cloud_liquid_water_content_dlat': 0.0, |
| 'specific_cloud_liquid_water_content_dlon': -0.0, |
| 'specific_humidity': 0.0, |
| 'specific_humidity_del2': 0.0, |
| 'specific_humidity_dlat': 0.0, |
| 'specific_humidity_dlon': 0.0, |
| 'surface_embedding': 0.0, |
| 'temperature_variation': -4.98, |
| 'temperature_variation_del2': 43.608, |
| 'temperature_variation_dlat': 4.843, |
| 'temperature_variation_dlon': 0.0, |
| 'u': 0.007, |
| 'u_del2': 0.242, |
| 'u_dlat': 0.002, |
| 'u_dlon': 0.0, |
| 'v': -0.0, |
| 'v_del2': -0.0, |
| 'v_dlat': -0.0, |
| 'v_dlon': 0.0, |
| 'vorticity': -0.002, |
| 'vorticity_del2': -0.365, |
| 'vorticity_dlat': 0.041, |
| 'vorticity_dlon': 0.0} |
|
|
| |
| |
| SigmaCoordinatesEquidistant.layers = %N_SIGMA_LAYERS |
|
|
| |
| |
| custom_corrds/SigmaCoordinatesEquidistant.layers = %N_SIGMA_LAYERS |
|
|
| |
| |
| advance/SoftClip.hinge_softness = 1.0 |
| advance/SoftClip.max_value = 16 |
| advance/SoftClip.name = None |
|
|
| |
| |
| StochasticModularStepModel.advance_module = @StochasticPhysicsParameterizationStep |
| StochasticModularStepModel.decoder_module = \ |
| @DimensionalLearnedPrimitiveToWeatherbenchDecoder |
| StochasticModularStepModel.encoder_module = \ |
| @DimensionalLearnedWeatherbenchToPrimitiveWithMemoryEncoder |
| StochasticModularStepModel.forcing_module = @DynamicDataForcing |
| StochasticModularStepModel.name = None |
|
|
| |
| |
| StochasticPhysicsParameterizationStep.checkpoint_substep = False |
| StochasticPhysicsParameterizationStep.corrector_module = %CORRECTOR_MODULE |
| StochasticPhysicsParameterizationStep.name = None |
| StochasticPhysicsParameterizationStep.num_substeps = %NUM_SUBSTEPS |
| StochasticPhysicsParameterizationStep.physics_parameterization_module = \ |
| @DivCurlNeuralParameterization |
| StochasticPhysicsParameterizationStep.randomness_module = @ZerosRandomField |
|
|
| |
| |
| advance/ToModalDiffOperators.name = None |
|
|
| |
| |
| encoder_data/ToModalDiffOperators.name = None |
|
|
| |
| |
| with_grads/ToModalDiffOperators.name = None |
|
|
| |
| |
| trajectory_from_step.checkpoint_multistep = False |
| trajectory_from_step.checkpoint_post_process = True |
| trajectory_from_step.checkpoint_step = True |
|
|
| |
| |
| advance/TruncateSigmaLevels.name = None |
| advance/TruncateSigmaLevels.sigma_ranges = \ |
| {'divergence': (0.0, 1), |
| 'geopotential': (0.0, 1), |
| 'specific_cloud_ice_water_content': (0.0, 1), |
| 'specific_cloud_liquid_water_content': (0.0, 1), |
| 'specific_humidity': (0.0, 1), |
| 't': (0.0, 1), |
| 'temperature_variation': (0.0, 1), |
| 'u': (0.0, 1), |
| 'u_component_of_wind': (0.0, 1), |
| 'v': (0.0, 1), |
| 'v_component_of_wind': (0.0, 1), |
| 'vorticity': (0.0, 1), |
| 'z': (0.0, 1)} |
|
|
| |
| |
| decoder_model/TruncateSigmaLevels.name = None |
| decoder_model/TruncateSigmaLevels.sigma_ranges = \ |
| {'divergence': (0.0, 1), |
| 'geopotential': (0.0, 1), |
| 'specific_cloud_ice_water_content': (0.0, 1), |
| 'specific_cloud_liquid_water_content': (0.0, 1), |
| 'specific_humidity': (0.0, 1), |
| 't': (0.0, 1), |
| 'temperature_variation': (0.0, 1), |
| 'u': (0.0, 1), |
| 'u_component_of_wind': (0.0, 1), |
| 'v': (0.0, 1), |
| 'v_component_of_wind': (0.0, 1), |
| 'vorticity': (0.0, 1), |
| 'z': (0.0, 1)} |
|
|
| |
| |
| advance/VelocityAndPrognostics.compute_gradients_module = @ToModalDiffOperators |
| advance/VelocityAndPrognostics.fields_to_include = \ |
| ['divergence', |
| 'vorticity', |
| 'u', |
| 'v', |
| 'temperature_variation', |
| 'log_surface_pressure', |
| 'specific_humidity', |
| 'specific_cloud_liquid_water_content', |
| 'specific_cloud_ice_water_content'] |
| advance/VelocityAndPrognostics.name = None |
|
|
| |
| |
| embedding_model/VelocityAndPrognostics.fields_to_include = \ |
| ['divergence', |
| 'vorticity', |
| 'u', |
| 'v', |
| 'temperature_variation', |
| 'specific_humidity', |
| 'specific_cloud_liquid_water_content', |
| 'specific_cloud_ice_water_content'] |
| embedding_model/VelocityAndPrognostics.name = None |
|
|
| |
| |
| encoder_data/VelocityAndPrognostics.compute_gradients_module = \ |
| @ToModalDiffOperators |
| encoder_data/VelocityAndPrognostics.fields_to_include = \ |
| ['u', |
| 'v', |
| 't', |
| 'z', |
| 'specific_humidity', |
| 'specific_cloud_liquid_water_content', |
| 'specific_cloud_ice_water_content'] |
| encoder_data/VelocityAndPrognostics.name = None |
|
|
| |
| |
| model/VelocityAndPrognostics.fields_to_include = None |
| model/VelocityAndPrognostics.name = None |
|
|
| |
| |
| VerticalConvTower.activate_final = False |
| VerticalConvTower.activation = %ACTIVATION |
| VerticalConvTower.channels = [64, 64, 64, 64] |
| VerticalConvTower.checkpoint_tower = True |
| VerticalConvTower.kernel_shape = 5 |
| VerticalConvTower.name = None |
| VerticalConvTower.with_bias = True |
|
|
| |
| |
| WhirlModel.from_xarray_fn = @xarray_to_state_and_dynamic_covariate_data |
| WhirlModel.model_cls = @StochasticModularStepModel |
| WhirlModel.to_xarray_fn = @data_to_xarray_with_renaming |
|
|
| |
| |
| xarray_to_data_with_renaming.renaming_dict = \ |
| {'geopotential': 'z', |
| 'latitude': 'lat', |
| 'longitude': 'lon', |
| 'temperature': 't', |
| 'u_component_of_wind': 'u', |
| 'v_component_of_wind': 'v'} |
| xarray_to_data_with_renaming.xarray_to_data_fn = @xarray_to_weatherbench_data |
|
|
| |
| |
| xarray_to_dynamic_covariate_data.covariates_to_include = \ |
| ('sea_ice_cover', 'sea_surface_temperature') |
|
|
| |
| |
| xarray_to_state_and_dynamic_covariate_data.values = 'values' |
| xarray_to_state_and_dynamic_covariate_data.xarray_to_dynamic_covariate_data_fn = \ |
| @xarray_to_dynamic_covariate_data |
| xarray_to_state_and_dynamic_covariate_data.xarray_to_state_data_fn = \ |
| @xarray_to_data_with_renaming |
|
|
| |
| |
| xarray_to_weatherbench_data.diagnostics_to_include = () |
| xarray_to_weatherbench_data.tracers_to_include = \ |
| ('specific_humidity', |
| 'specific_cloud_liquid_water_content', |
| 'specific_cloud_ice_water_content') |
|
|
| |
| |
| ZerosRandomField.prefer_nodal = True |
|
|
