一个 Python 脚本中的多个 TFLearn 模型答案

【问题标题】：Multiple TFLearn Models in One Python Script一个 Python 脚本中的多个 TFLearn 模型
【发布时间】：2017-11-18 03:27:33
【问题描述】：

我在单个 python3 脚本中按顺序运行多个神经网络模型，如下所示：

neural_net_model = tflearn.DNN(network, checkpoint_path=XYZ, 
                    tensorboard_dir=DIR), tensorboard_verbose=2, max_checkpoints=1)

第一个模型运行良好。当我实例化第二个模型时，我总是会得到这个错误。

InvalidArgumentError：形状 [-1,224,224,1] 具有负尺寸

224x224 是我正在处理的图像的尺寸

如果我重新启动内核，模型运行良好

是否可以明确关闭或清除第一个模型，以便新模型顺利运行？

我在 Anaconda 环境中运行

【问题讨论】：

【解决方案1】：

以下代码将重置图形并允许脚本运行新模型

tf.reset_default_graph()

【讨论】：