问题描述:
在运行霹雳吧啦Wz老师用tensorflow实现的Alexnet代码时,出现以下问题:
问题一:train.py脚本中:
train_accuracy = history_dict["accuracy"]
KeyError: ‘accuracy’
问题二:trainGPU.py脚本中:
raise RuntimeError("__iter__() is only supported inside of tf.function "
RuntimeError: __iter__() is only supported inside of tf.function or when eager execution is enabled.
原因分析:
问题一:由于keras库版本不同所造成的
问题二:使用动态图机制时,tensorflow版本问题造成的。
tensorflow2.0以上版本是默认打开动态图机制的,tensorflow 2.0 以下的版本,需要调用 tf.enable_eager_execution() 来启动动态图机制