我正在尝试训练一个模型来识别不同的云类型并选择那些云所在的区域。为此,我将图像文件(350 * 550,灰度)发送到模型,该模型输出要选择的像素(350 * 550)。我在下面的代码中附加了一个链接。这是与Kaggle(https://www.kaggle.com/c/understanding_cloud_organization/data)竞争的一部分,我不希望有人重写我的整个代码。
当尝试训练我的数据时,数据会被加载,然后发送到模型以适合。然后我从tensorflows bfc分配中获得了一堆代码,它在几分钟后退出,没有明显的错误(至少对我来说)。我真的不知所措。
代码:https://github.com/abritton99999999/KaggleCode 控制台输出现在也发布在github中!
答案 0 :(得分:0)
您正在尝试创建一个非常大的张量(50000x192500浮点数),并且超出了内存限制。一些解决方法的建议在这里 Tensorflow Error: "Cannot parse tensor from proto"
基本上,您将无法创建这样的网络
model.add(Dense(100000, input_shape=(192500,), activation="sigmoid"))
model.add(Dense(50000, activation="sigmoid"))
model.add(Dense(192500, activation="softmax"))
相反,您需要使用卷积层