temp fix ema

google · Nov 22, 2020 · c64759c · c64759c
1 parent 4c83fbe
commit c64759c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/efficientdet/keras/train_lib.py b/efficientdet/keras/train_lib.py
@@ -398,7 +398,7 @@ def _draw_inference(self, step):
 
 def get_callbacks(params, val_dataset):
   """Get callbacks for given params."""
-  if params.get('moving_average_decay', None):
+  if False:
     from tensorflow_addons.callbacks import AverageModelCheckpoint
     avg_callback = AverageModelCheckpoint(
         filepath=os.path.join(params['model_dir'], 'ckpt'),