在一个Python脚本中运行多个TFLearn模型

我正在一个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

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

如果我重启内核,模型就能正常运行

是否可以显式地关闭或清除第一个模型,以便新模型能够顺利运行?

我是在Anaconda环境中运行的


回答:

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

tf.reset_default_graph()

Related Posts

L1-L2正则化的不同系数

我想对网络的权重同时应用L1和L2正则化。然而,我找不…

使用scikit-learn的无监督方法将列表分类成不同组别,有没有办法?

我有一系列实例,每个实例都有一份列表,代表它所遵循的不…

f1_score metric in lightgbm

我想使用自定义指标f1_score来训练一个lgb模型…

通过相关系数矩阵进行特征选择

我在测试不同的算法时,如逻辑回归、高斯朴素贝叶斯、随机…

可以将机器学习库用于流式输入和输出吗?

已关闭。此问题需要更加聚焦。目前不接受回答。 想要改进…

在TensorFlow中,queue.dequeue_up_to()方法的用途是什么?

我对这个方法感到非常困惑,特别是当我发现这个令人费解的…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注