一开始百度都在说参数初始化的问题,后面检查代码发现在训练Inceptionv3模型的时候,给的参数是Inceptionv4的,就出现了这个问题
with tf.Session(config=tf.ConfigProto(allow_soft_placement=True)) as sess:
#s1.restore(sess, model_checkpoint_map['inception_v3'])
s2.restore(sess, model_checkpoint_map['inception_v4'])
#s3.restore(sess, model_checkpoint_map['inception_resnet_v2'])
#s4.restore(sess, model_checkpoint_map['resnet_v2'])
# s5.restore(sess, model_checkpoint_map['ens3_adv_inception_v3'])
# s6.restore(sess, model_checkpoint_map['ens4_adv_inception_v3'])
# s7.restore(sess, model_checkpoint_map['ens_adv_inception_resnet_v2'])
# s8.restore(sess, model_checkpoint_map['adv_inception_v3'])