search

Tencent / tencent-ml-images

Largest multi-label image database; ResNet-101 model; 80.73% top-1 acc on ImageNet
Other
3.06k stars 515 forks source link

Finetune Error #35

Open 0xuye0 opened 5 years ago

0xuye0 commented 5 years ago

finetune报错如下 image 尝试了好久没有解决。。不知道会是何原因呢。。

我的环境是python3+tensorflow1.10.1

0xuye0 commented 5 years ago

补充一下，是在finetune.py的 _, step, loss = sess.run([train_op, global_step, Loss]) 这一行报错

AIML commented 5 years ago

看你错误的位置是你的tfrecord解析脚本中feature的定义有问题；请查看示例代码中tfrecord的生成脚本。label是一个one_hot类型的vector，被编码成了一个string，要使用decode_raw解码出来

scriptboy1990 commented 5 years ago

finetune报错如下 image 尝试了好久没有解决。。不知道会是何原因呢。。

我的环境是python3+tensorflow1.10.1

请问解决了这个问题了吗。

Maggie-rhino commented 5 years ago

看你错误的位置是你的tfrecord解析脚本中feature的定义有问题；请查看示例代码中tfrecord的生成脚本。label是一个one_hot类型的vector，被编码成了一个string，要使用decode_raw解码出来

it works！！