qq_36556054的博客

原创 tf.data.Dataset读取TFRecord文件进行训练

前言 之前跑了一个小的CNN，因为训练数据量也不大，自己的电脑显卡2g，直接将数据读到内存中再取batch来训练还能直接跑；后来数据集大了一些，再用这种方法就不行了，内存不足。没办法，找了一些TFRecord和Dataset的资料把程序改了改，这样就可以边取数据边训练，总算能跑了，以后再也不用怕训练数据多啦。 数据 我的训练数据是一些元素为float类型的矩阵，用numpy生成一些来代表吧： da...