minor...

modules/deeplearning/icing_fcn.py

@@ -1001,6 +1001,7 @@ class IcingIntensityFCN:
 
         ckpt.restore(ckpt_manager.latest_checkpoint)
 
+        if TRACK_MOVING_AVERAGE:
             for idx, var in enumerate(self.model.trainable_variables):
                 var.assign(self.ema_trainable_variables[idx])