用tensorflow编写网络模型时内存不够

尽量用稀疏矩阵,减少batch大小
■网友
告诉你一个祖传秘方:别用那么大的矩阵
■网友
尽量用Python的迭代器,或者用tf record读取数据,batchsize改小。不想改代码的租个gpu云服务器,4,5块一小时吧。
■网友
不改模型你就减少batch size吧
■网友
最简单粗暴的方法是把这一层丢到cpu里面去,就不占用显存了


    推荐阅读