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

Question

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

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

第一个模型运行没问题。当我实例化第二个模型时，我总是会收到此错误。

InvalidArgumentError: Shape [-1,224,224,1] has negative dimensions

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

如果我重新启动内核，模型 运行 没问题

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

我运行正在使用 Anaconda 环境

Answer 1

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

tf.reset_default_graph()