From ca3280ca9463e947ccb2b5695c3cbc74fa0f6723 Mon Sep 17 00:00:00 2001 From: Damian Rouson Date: Mon, 9 Feb 2026 15:52:23 -0800 Subject: [PATCH] feat(training_config): expand tensor components This commit updates the "tensor names" field in the training_configuration.json file to add additional the following input/output variables for training: ni, nr, qi, and qg. --- demo/training_configuration.json | 6 +++--- demo/training_data_files.json | 5 +++++ 2 files changed, 8 insertions(+), 3 deletions(-) diff --git a/demo/training_configuration.json b/demo/training_configuration.json index 367daa6fb..c31d7b186 100644 --- a/demo/training_configuration.json +++ b/demo/training_configuration.json @@ -6,11 +6,11 @@ }, "network configuration": { "skip connections" : false, - "nodes per layer" : [7,32,32,32,5], + "nodes per layer" : [11,32,32,32,9], "activation function" : "sigmoid" }, "tensor names": { - "inputs" : ["potential_temperature","qv", "qc", "qr", "qs", "pressure", "temperature"], - "outputs" : ["potential_temperature","qv", "qc", "qr", "qs"] + "inputs" : ["potential_temperature","qv", "qc", "qr", "qs", "pressure", "temperature", "ni", "nr", "qi", "qg"], + "outputs" : ["potential_temperature","qv", "qc", "qr", "qs", "ni", "nr", "qi", "qg"] } } diff --git a/demo/training_data_files.json b/demo/training_data_files.json index d88ffee0b..39318c9bb 100644 --- a/demo/training_data_files.json +++ b/demo/training_data_files.json @@ -6,3 +6,8 @@ "infixes" : ["450", "500"] } } + +training_input-image-_450.nc +training_output-image-_450.nc +training_input-image-_500.nc +training_output-image-_500.nc