2019年11月7日
tensorflow 函数功能集锦
摘要： tf.add_to_collection（） tf.get_collection() tf.add_n() tf.add_to_collection：把变量放入一个集合，把很多变量变成一个列表 tf.get_collection：从一个结合中取出全部变量，是一个列表 tf.add_n：把一个列表的东 阅读全文
posted @ 2019-11-07 18:59 闪存第一菜鸡 阅读(180) 评论(0) 推荐(0)
tensorflow ResourceExhaustedError (see above for traceback): OOM when allocating tensor with shape[256,256,15,15] and type float on 报错（显存不够报错）
摘要： 出现这个的原因是gpu显存不够导致的，一般是我们设置了程序根据需求增长导致的 这里，我们设置 allow_growth=False 就可以控制显存使用的增长，从而控制显存的使用，而不会程序运行一半报错。 这种情况是建立在我们的显存比较小的情况下使用的策略，如果说我们显存够大则不用担心这种问题。 阅读全文
posted @ 2019-11-07 18:50 闪存第一菜鸡 阅读(1142) 评论(0) 推荐(0)