matlab - 使用 MNIST 数据集运行“堆叠自动编码器示例”程序（深度学习工具箱）时出现 Matlab“内存不足”错误

Question

我是深度学习的新手，我使用的是 matlab 的深度学习工具箱。我想运行：“test_example_SAE.m”，它构建了一个堆叠的自动编码器并使用 MNIST 数据集对其进行训练和测试，但由于这个错误我不能：*

Error using horzcat
Out of memory. Type HELP MEMORY for your options.

* 这份工作需要多少内存？我的意思是我可以在具有 4GB RAM 的普通 PC 上运行深度学习工具箱代码吗？还是我应该学习在 GPU 上运行代码？

Answer 1

它也发生在我身上。如果您可以减少样本数量。

此外，如果您正在运行“for”循环，请尝试将其更改为向量。

另一个提示是将数据分成几部分（如果可能的话）并尝试逐个运行。

这些只是我的建议。如果你在 GPU 上运行它没有区别。

如果您的工作或大学可用，您也可以尝试使用集群计算。