Traceback (most recent call last):
File "keras_retinanet/bin/train_vocab_w2v.py", line 407, in
main()
File "keras_retinanet/bin/train_vocab_w2v.py", line 361, in main
training_model.load_weights('Model/resnet50_csv_50_focal_seen_w2v.h5')
File "/usr/local/miniconda3/envs/keras/lib/python3.6/site-packages/keras/engine/network.py", line 1166, in load_weights
f, self.layers, reshape=reshape)
File "/usr/local/miniconda3/envs/keras/lib/python3.6/site-packages/keras/engine/saving.py", line 1045, in load_weights_from_hdf5_group
reshape=reshape)
File "/usr/local/miniconda3/envs/keras/lib/python3.6/site-packages/keras/engine/saving.py", line 682, in preprocess_weights_for_loading
weights = convert_nested_model(weights)
File "/usr/local/miniconda3/envs/keras/lib/python3.6/site-packages/keras/engine/saving.py", line 658, in convert_nested_model
original_backend=original_backend))
File "/usr/local/miniconda3/envs/keras/lib/python3.6/site-packages/keras/engine/saving.py", line 801, in preprocess_weights_for_loading
weights[0] = np.transpose(weights[0], (3, 2, 0, 1))
File "/usr/local/miniconda3/envs/keras/lib/python3.6/site-packages/numpy/core/fromnumeric.py", line 598, in transpose
return _wrapfunc(a, 'transpose', axes)
File "/usr/local/miniconda3/envs/keras/lib/python3.6/site-packages/numpy/core/fromnumeric.py", line 51, in _wrapfunc
return getattr(obj, method)(*args, **kwds)
ValueError: axes don't match array