Internal change

PiperOrigin-RevId: 423362449

Internal change
PiperOrigin-RevId: 423362449
ab8b8012 · A. Unique TensorFlower · de44470f · ab8b8012
Commit ab8b8012 authored Jan 21, 2022 by A. Unique TensorFlower
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

official/core/actions.py official/core/actions.py +1 -1

No files found.
--- a/official/core/actions.py
+++ b/official/core/actions.py
@@ -201,7 +201,7 @@ def get_train_actions(
  """Gets train actions for TFM trainer."""
  train_actions = []
  # Adds pruning callback actions.
-  if hasattr(params.task, 'pruning'):
+  if hasattr(params.task, 'pruning') and params.task.pruning:
    train_actions.append(
        PruningAction(
            export_dir=model_dir,