From b41cf7495893e76709f5583759cfdc9166afd78e Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Wed, 15 Dec 2021 10:08:15 -0600
Subject: [PATCH] snapshot...
---
modules/deeplearning/icing_cnn.py | 7 ++++---
1 file changed, 4 insertions(+), 3 deletions(-)
diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py
index 98493282..0418dd8d 100644
--- a/modules/deeplearning/icing_cnn.py
+++ b/modules/deeplearning/icing_cnn.py
@@ -8,7 +8,6 @@ import numpy as np
import pickle
import h5py
-USE_FLIGHT_ALTITUDE = False
LOG_DEVICE_PLACEMENT = False
@@ -117,7 +116,7 @@ def build_residual_block(input, drop_rate, num_neurons, activation, block_name,
class IcingIntensityNN:
- def __init__(self, day_night='DAY', l1b_andor_l2='both', gpu_device=0, datapath=None):
+ def __init__(self, day_night='DAY', l1b_andor_l2='both', use_flight_altitude=False, gpu_device=0, datapath=None):
if day_night == 'DAY':
self.train_params_l1b = train_params_l1b_day
@@ -233,6 +232,8 @@ class IcingIntensityNN:
self.DISK_CACHE = False
+ self.USE_FLIGHT_ALTITUDE = use_flight_altitude
+
if datapath is not None:
self.DISK_CACHE = False
f = open(datapath, 'rb')
@@ -928,7 +929,7 @@ class IcingIntensityNN:
# flat = tf.keras.layers.concatenate([flat, flat_1d, flat_anc])
# flat = tf.keras.layers.concatenate([flat, flat_1d])
# self.build_dnn(flat)
- if USE_FLIGHT_ALTITUDE:
+ if self.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