想要将tf.tensor转换为numpy格式,网上介绍的无非就是两种方法,一种是直接xxx.numpy(),结果报错,按照网上说的修改方法也不行;另一种方法就是使用session,然后xxxxx.eval(session=sess),结果会导致内存直接爆炸,16g的内存装满,显示的warning为
2021-03-16 00:22:36.412770: W tensorflow/core/framework/cpu_allocator_impl.cc:80] Allocation of 4759040000 exceeds 10% of free system memory.
2021-03-16 00:22:37.746183: W tensorflow/core/framework/cpu_allocator_impl.cc:80] Allocation of 4055040000 exceeds 10% of free system memory.
代码为
with tf.compat.v1.Session() as sess:
sess.run(tf.compat.v1.global_variables_initializer())
training_data_x = training_data_x.eval(session=sess)
其中training_data_x为tf.tensor的矩阵,想转化为numpy
求求大佬们帮帮孩子吧,有没有别的数据转化方式或者有没有办法解决这个内存爆炸的问题