From 7d712fabe0313005de90ee515888d9bd523d1d45 Mon Sep 17 00:00:00 2001
From: Neta Zmora <neta.zmora@intel.com>
Date: Tue, 13 Aug 2019 22:18:15 +0300
Subject: [PATCH] AMC: rewards.py - shorten log statement
---
examples/auto_compression/amc/rewards.py | 7 ++-----
1 file changed, 2 insertions(+), 5 deletions(-)
diff --git a/examples/auto_compression/amc/rewards.py b/examples/auto_compression/amc/rewards.py
index 480ee1b..ab16cb4 100755
--- a/examples/auto_compression/amc/rewards.py
+++ b/examples/auto_compression/amc/rewards.py
@@ -78,11 +78,8 @@ def mac_constrained_clamp_action(env, pruning_action):
duty/env.original_model_macs,
flops/env.original_model_macs))
if pruning_action_final != pruning_action:
- msglogger.debug("action ********** pruning_action={}==>pruning_action_final={:.2f}: reduced={:.2f} rest={:.2f} target={:.2f} duty={:.2f} flops={:.2f}".
- format(pruning_action, pruning_action_final, reduced/env.original_model_macs,
- rest/env.original_model_macs, 1-env.amc_cfg.target_density,
- duty/env.original_model_macs,
- flops/env.original_model_macs))
+ msglogger.debug("pruning_action={:.2f}==>pruning_action_final={:.2f}".format(pruning_action,
+ pruning_action_final))
return pruning_action_final
--
GitLab