From 9ffbd4d8c7976f66230f2e33dc49ee45b569f4cf Mon Sep 17 00:00:00 2001 From: Silvan Laube Date: Wed, 8 Jul 2020 11:51:59 +0200 Subject: [PATCH] bugfix preventing to load data It seems the default value for allow_pickle changed along the way. The repo could also use support for TF2, should not be much effort if this works: https://stackoverflow.com/a/59027965/540725 --- kaffe/tensorflow/network.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/kaffe/tensorflow/network.py b/kaffe/tensorflow/network.py index 3f7f023..62cba51 100644 --- a/kaffe/tensorflow/network.py +++ b/kaffe/tensorflow/network.py @@ -56,7 +56,7 @@ def load(self, data_path, session, ignore_missing=False): session: The current TensorFlow session ignore_missing: If true, serialized weights for missing layers are ignored. ''' - data_dict = np.load(data_path).item() + data_dict = np.load(data_path, allow_pickle=True).item() for op_name in data_dict: with tf.variable_scope(op_name, reuse=True): for param_name, data in data_dict[op_name].iteritems():