Models: InvalidArgumentError when fine-tuning inception_v3 on flowers

Created on 5 Nov 2016  路  3Comments  路  Source: tensorflow/models

I'm running TensorFlow 0.11.0rc0 in ubuntu 14.04.

When I tried to fine-tuning Inception_v3 on flowers dataset follow that command:
python train_image_classifier.py
--train_dir=${TRAIN_DIR}
--dataset_dir=${DATASET_DIR}
--dataset_name=flowers
--dataset_split_name=train
--model_name=inception_v3
--checkpoint_path=${CHECKPOINT_PATH}
--checkpoint_exclude_scopes=InceptionV3/Logits,InceptionV3/AuxLogits/Logits
--trainable_scopes=InceptionV3/Logits,InceptionV3/AuxLogits/Logits

It shows the errors:
InvalidArgumentError (see above for traceback): Assign requires shapes of both tensors to match. lhs shape= [1,1,768,5] rhs shape= [1,1,768,1001]

I thought the flags ----checkpoint_exclude_scopes will avoid that unbalance shapes.
Thanks

help wanted

Most helpful comment

All 3 comments

This is still outstanding and I've tested that the pull request #632 works. @girving or @sguada is it possible to merge #632 and close this issue?

Was this page helpful?
0 / 5 - 0 ratings

Related issues

aifollower picture aifollower  路  49Comments

Tsuihao picture Tsuihao  路  90Comments

10183308 picture 10183308  路  50Comments

NilsLattek picture NilsLattek  路  61Comments

DanMossa picture DanMossa  路  48Comments