From d51cc71c599ccdf4f89ad213335ce0087af3ee48 Mon Sep 17 00:00:00 2001 From: tomrink <rink@ssec.wisc.edu> Date: Mon, 25 Oct 2021 10:51:00 -0500 Subject: [PATCH] add switch for using/not using flight altitude: this will help us merge with the master branch --- modules/deeplearning/icing_cnn.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py index 2fd21c81..0afe586c 100644 --- a/modules/deeplearning/icing_cnn.py +++ b/modules/deeplearning/icing_cnn.py @@ -2,13 +2,14 @@ import tensorflow as tf import tensorflow_addons as tfa from util.setup import logdir, modeldir, cachepath, now from util.util import homedir, EarlyStop, normalize, make_for_full_domain_predict -from util.geos_nav import GEOSNavigation, get_navigation +from util.geos_nav import get_navigation import os, datetime import numpy as np import pickle import h5py +USE_FLIGHT_ALTITUDE = True LOG_DEVICE_PLACEMENT = False @@ -910,7 +911,8 @@ class IcingIntensityNN: # flat = tf.keras.layers.concatenate([flat, flat_1d, flat_anc]) # flat = tf.keras.layers.concatenate([flat, flat_1d]) # self.build_dnn(flat) - flat = tf.keras.layers.concatenate([flat, self.inputs[1]]) + if USE_FLIGHT_ALTITUDE: + flat = tf.keras.layers.concatenate([flat, self.inputs[1]]) self.build_dnn(flat) self.model = tf.keras.Model(self.inputs, self.logits) -- GitLab