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