snapshot...

modules/deeplearning/icing_fcn.py

@@ -719,7 +719,7 @@ class IcingIntensityFCN:
             # optimizer = tfa.optimizers.MovingAverage(optimizer)
             self.ema = tf.train.ExponentialMovingAverage(decay=0.9999)
             self.ema.apply(self.model.trainable_variables)
-            self.ema_trainable_variables = self.ema.average(self.model.trainable_variables)
+            self.ema_trainable_variables = [self.ema.average(var) for var in self.model.trainable_variables]
 
         self.optimizer = optimizer
         self.initial_learning_rate = initial_learning_rate