For compatibility with Python3 unpickling has to be done in "bytes" mode (default is "ascii")
@@ -39,7 +39,7 @@ def model_name():
def load_weights(data_path):
global WEIGHTS
- WEIGHTS = np.load(data_path).item()
+ WEIGHTS = np.load(data_path, encoding='bytes').item()
#-----------------------------------------------------------------------
