From 725a89f4e400049be4530630d8a94632b9ad66d0 Mon Sep 17 00:00:00 2001
From: Neta Zmora <neta.zmora@intel.com>
Date: Wed, 20 Feb 2019 17:33:08 +0200
Subject: [PATCH] scheduler: backward compatibility for the signature of
load_state_dict
Add a default value for the new parameter 'normalize_dataparallel_keys'
---
distiller/scheduler.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/distiller/scheduler.py b/distiller/scheduler.py
index a5e360d..a24de1a 100755
--- a/distiller/scheduler.py
+++ b/distiller/scheduler.py
@@ -189,7 +189,7 @@ class CompressionScheduler(object):
state = {'masks_dict': masks}
return state
- def load_state_dict(self, state, normalize_dataparallel_keys):
+ def load_state_dict(self, state, normalize_dataparallel_keys=False):
"""Loads the scheduler state.
Currently the scheduler state is comprised only of the set of pruning masks.
--
GitLab