search

titu1994 / keras-efficientnets

Keras Implementation of EfficientNets
MIT License
187 stars 63 forks source link

Wrong parameter example #16

Closed FedericoRaimondi closed 5 years ago

FedericoRaimondi commented 5 years ago

there is a little typo in the readme example

model = EfficientNetB0(input_size, classes=1000, include_top=True, weights='imagenet')

should be

model = EfficientNetB0(input_shape, classes=1000, include_top=True, weights='imagenet')
titu1994 commented 5 years ago

It's minor. I'll fix it in the next release. Thanks for bringing it to my attention.