我有 6000 张图像要在神经网络上进行训练。
我目前的 PC 规格:-
32GB RAM,i5 2 核处理器,标准 GPU(GPU 上没有工作),1TB 硬盘
我的神经网络规格:-
3000 Epoch 大小,6 Batch 大小
这些规范正在杀死进程。
训练好它需要多少硬件?理想的 Batch Size 和 Epoch Size 是多少?
我有 6000 张图像要在神经网络上进行训练。
我目前的 PC 规格:-
32GB RAM,i5 2 核处理器,标准 GPU(GPU 上没有工作),1TB 硬盘
我的神经网络规格:-
3000 Epoch 大小,6 Batch 大小
这些规范正在杀死进程。
训练好它需要多少硬件?理想的 Batch Size 和 Epoch Size 是多少?
这不是一个真正的数据科学特定问题,所以你可能想在其他地方问;无论如何,您需要提供更多信息!
您收到什么错误消息?你的内存不足了吗?图像尺寸是多少?
除非您的图像真的很大,否则我不希望您的内存不足,因为您有 32Gb。然而,如果你建立了一个庞大的网络,例如许多 ConvNet 过滤器然后被传递到一个非常宽的全连接层(密集层),你将拥有大量的权重,这可能会导致内存错误。
尝试使用 Keras 而不是纯 Tensorflow 创建您的模型(或类似模型),然后model.summary
在编译后的模型上使用。这将向您展示每层中权重数量的一个很好的概览。