Aardvark-Weather / config.json
yzt15806542928's picture
新增config.json文件
3397141 verified
Raw History Blame Contribute Delete
1.24 kB
{
"model_name": "Aardvark Weather",
"model_type": "aardvark-weather",
"architectures": [
"ConvCNPWeatherE2E"
],
"framework": "PyTorch",
"domain": "atmosphere",
"task": "end-to-end-multimodal-weather-forecasting",
"implementation": {
"entry_point": "model/aardvark_adapter.py",
"scope": "official encoder, one-day forecast processor, and station-temperature decoder"
},
"architecture": {
"stages": [
"observation_encoder",
"global_forecast_processor",
"station_decoder"
],
"encoder_input_channels": 277,
"encoder_output_channels": 24,
"processor_input_channels": 35,
"processor_output_channels": 24,
"tas_decoder_input_channels": 36,
"tas_decoder_output_channels": 1
},
"forecasting": {
"lead_time_days": 1,
"global_grid_shape": [
121,
240
],
"global_grid_resolution_degrees": 1.5,
"global_state_channels": 24,
"station_variable": "2m_temperature",
"station_count": 8719
},
"configuration_sources": [
"README.md",
"official-src/aardvark/e2e_model.py",
"weights/trained_model/encoder/config.pkl",
"weights/trained_model/processor/config.pkl",
"weights/trained_model/decoder/tas/config.pkl"
]
}