The problem of fine-tuning NASNet-A_Large model

yeephycho / nasnet-tensorflow

A nasnet in tensorflow

Apache License 2.0

136 stars 52 forks source link

The problem of fine-tuning NASNet-A_Large model #12

Closed TTMRonald closed 6 years ago

TTMRonald commented 6 years ago

Hi, yeephycho. I meet a problem with NASNet-A_Large pretrained model. I hope to get your help. Thanks. INFO:tensorflow:Error reported to Coordinator: <class 'tensorflow.python.framework.errors_impl.InvalidArgumentError'>, Assign requires shapes of both tensors to match. lhs shape= [751] rhs shape= [1001]

yeephycho commented 6 years ago

You need to change the number of classes and please make sure your input image size is correct.