Skip to content

GitLab

Commit a2fc0302 authored by A. Unique TensorFlower's avatar A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 423362449
parent b482319b
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/core/actions.py
View file @ a2fc0302
...@@ -201,7 +201,7 @@ def get_train_actions( ...@@ -201,7 +201,7 @@ def get_train_actions(
"""Gets train actions for TFM trainer.""" """Gets train actions for TFM trainer."""
train_actions = [] train_actions = []
# Adds pruning callback actions. # Adds pruning callback actions.
if hasattr(params.task, 'pruning'): if hasattr(params.task, 'pruning') and params.task.pruning:
train_actions.append( train_actions.append(
PruningAction( PruningAction(
export_dir=model_dir, export_dir=model_dir,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment