尝试使用新类重新训练最后一层初始v3模型。我准备了自己的一组图像并存储在C:\ tmp中。当我运行代码时
C:\tmp\image_retrain\tfClassifier\image_classification>python retrain.py --model_dir=C:\tmp\image_retrain\tfClassifier\image_classification\inception --image_dir=C:\tmp\image_retrain\tfClassifier\image_classification\styles --output_graph=C:\tmp\image_retrain\tfClassifier\image_classification\output_dir\all_ouput_stored_here.pb --how_many_training_steps 500
它启动分类并创建瓶颈文件,但是我得到了这个错误,我无法弄清楚是什么问题
File "retrain.py", line 967, in <module>
tf.app.run(main=main, argv=[sys.argv[0]] + unparsed)
File "C:\Users\aamo8\Anaconda3\envs\tensorflow-gpu\lib\site-packages\tensorflow\python\platform\app.py", line 48, in run
_sys.exit(main(_sys.argv[:1] + flags_passthrough))
File "retrain.py", line 750, in main
train_writer = tf.train.SummaryWriter(FLAGS.summaries_dir + '/train',
AttributeError: module 'tensorflow.python.training.training' has no attribute 'SummaryWriter'