S676951 2017-12-05 13:00 采纳率: 100%
浏览 8084
已采纳

如何解决Tensorflow数据集太大的问题?

将稀疏矩阵压缩时用的tensorflow.sparse_to_tense函数,报错Cannot create a tensor proto whose content is larger than 2GB,这个问题要怎么解决啊?老师给我们的数据集最大的一个有10GB,才接触TensorFlow,不知道怎么处理,麻烦大神们帮帮忙!

  • 写回答

1条回答 默认 最新

  • COCO_AS 2017-12-05 13:42
    关注

    这里有解决办法
    https://stackoverflow.com/questions/38087342/use-large-dataset-in-tensorflow

    搬过来供你参考

    Do not load data to constant, it will be part of your computational graph.

    You should rather:

    Create an op which is loading your data in stream fashion
    Load data in python part, and use feed_dict to pass the batch into the graph
    

    For TensorFlow 1.x and Python 3, there is my simple solution:

     X_init = tf.placeholder(tf.float32, shape=(m_input, n_input))
    X = tf.Variable(X_init)
    sess.run(tf.global_variables_initializer(), feed_dict={X_init: data_for_X})
    
    

    In practice, you will mostly specify Graph and Session for continuous computation, this following code will help you:

     my_graph = tf.Graph()
    sess = tf.Session(graph=my_graph)
    with my_graph.as_default():
        X_init = tf.placeholder(tf.float32, shape=(m_input, n_input))
        X = tf.Variable(X_init)
        sess.run(tf.global_variables_initializer(), feed_dict={X_init: data_for_X})
        .... # build your graph with X here
    .... # Do some other things here
    with my_graph.as_default():
        output_y = sess.run(your_graph_output, feed_dict={other_placeholder: other_data})
    
    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论

报告相同问题?

问题事件

  • 已采纳回答 9月16日

悬赏问题

  • ¥15 给我一个openharmony跑通webrtc实现视频会议的简单demo项目,sdk为12
  • ¥15 vb6.0使用jmail接收smtp邮件并另存附件到D盘
  • ¥30 vb net 使用 sendMessage 如何输入鼠标坐标
  • ¥15 关于freesurfer使用freeview可视化的问题
  • ¥100 谁能在荣耀自带系统MagicOS版本下,隐藏手机桌面图标?
  • ¥15 求SC-LIWC词典!
  • ¥20 有关esp8266连接阿里云
  • ¥15 C# 调用Bartender打印机打印
  • ¥15 我这个代码哪里有问题 acm 平台上显示错误 90%,我自己运行好像没什么问题
  • ¥50 C#编程中使用printDocument类实现文字排版打印问题