ValueError: Dimension 0 in both shapes must be equal, but are 7 and 64. Shapes are [7,7,448,64] and [64,3,7,7]. for 'Assign' (op: 'Assign') with input shapes: [7,7,448,64], [64,3,7,7].

iamhankai / attribute-aware-attention

[ACM MM 2018] Attribute-Aware Attention Model for Fine-grained Representation Learning

https://arxiv.org/abs/1901.00392

156 stars 30 forks source link

Closed zz10001 closed 4 years ago

zz10001 commented 5 years ago

when i run run.sh.it happened.what should i do?

iamhankai commented 5 years ago

It seemss that one of the input has wrong shape. 64 is the batch size, it should at the 1st dimension.

iamhankai commented 5 years ago

I guess you should change "image_data_format" in $HOME/.keras/keras.json

zz10001 commented 5 years ago

thx，i will try it。