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():