diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py
index f7c464e44b8bc6d90f3a82c68a3be0f2f0f5712f..3c0336342745d914859f0b5e6cac868548b7896a 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)