courses icon indicating copy to clipboard operation
courses copied to clipboard

move model.flatten above so that model compiles

Open vinbhaskara opened this issue 7 years ago • 1 comments

Currently, the model wouldn't compile for sizes other than (224,224) when fine tuning as under the branch if not include_top, the model is not yet flattened but later in the create function, a dense layer is added (without flattening), hence, raising an error. Changing the position of the model.flatten fixes this.

vinbhaskara avatar Jun 11 '17 17:06 vinbhaskara

Can you please confirm this works correctly in keras 1?

jph00 avatar Aug 03 '17 00:08 jph00