From c6b1dbb53f692c77f90d4d79ff4568664739dd61 Mon Sep 17 00:00:00 2001
From: Neta Zmora <neta.zmora@intel.com>
Date: Sat, 30 Jun 2018 00:55:49 +0300
Subject: [PATCH] Bug fix: compression schedule configuration parsing
Used the wrong indentation when parsing RegularizationPolicy
---
distiller/config.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/distiller/config.py b/distiller/config.py
index a5cc9cd..0e2fd62 100755
--- a/distiller/config.py
+++ b/distiller/config.py
@@ -82,7 +82,7 @@ def dict_config(model, optimizer, sched_dict):
assert instance_name in regularizers, "Regularizer {} was not defined in the list of regularizers".format(instance_name)
regularizer = regularizers[instance_name]
if args is None:
- policy = distiller.RegularizationPolicy(regularizer)
+ policy = distiller.RegularizationPolicy(regularizer)
else:
policy = distiller.RegularizationPolicy(regularizer, **args)
--
GitLab