From 2ba2abeb8356a88aad4265b68486ebbe927e46e8 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Thu, 29 Apr 2021 23:21:52 -0500
Subject: [PATCH] snapshot...
---
modules/deeplearning/icing_cnn.py | 5 ++---
1 file changed, 2 insertions(+), 3 deletions(-)
diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py
index f7c464e4..3c033634 100644
--- a/modules/deeplearning/icing_cnn.py
+++ b/modules/deeplearning/icing_cnn.py
@@ -1,8 +1,7 @@
import tensorflow as tf
-import tensorflow_addons as tfa
+#import tensorflow_addons as tfa
from util.setup import logdir, modeldir, cachepath
from util.util import homedir
-import subprocess
import os, datetime
import numpy as np
@@ -401,7 +400,7 @@ class IcingIntensityNN:
self.learningRateSchedule = tf.keras.optimizers.schedules.ExponentialDecay(initial_learning_rate, decay_steps, decay_rate)
optimizer = tf.keras.optimizers.Adam(learning_rate=self.learningRateSchedule)
- optimizer = tfa.optimizers.MovingAverage(optimizer)
+ # optimizer = tfa.optimizers.MovingAverage(optimizer)
if TRACK_MOVING_AVERAGE:
ema = tf.train.ExponentialMovingAverage(decay=0.999)
--
GitLab