tensorflow延迟初始化局部变量

2024-05-14 17:30:44 发布

您现在位置:Python中文网/ 问答频道 /正文

我的数据库是一个嵌套的图像文件夹。我在用操作系统步行切片输入生产者,以创建一个文件名队列,它被反馈给整个文件读取器。

代码在我的数据库的一小部分上工作良好(当爬网速度很快时)。但在整个错误上抛出“超出范围”。对于较大的数据库,可以通过将“无”设置“纪元计数器”来解决此问题。

是否有方法使用延迟/依赖项初始化变量?

代码的最小示例如下:

ligand_file_list,receptor_file_list = index_the_database(database_path)

ligand_files = tf.convert_to_tensor(ligand_file_list,dtype=tf.string)
receptor_files = tf.convert_to_tensor(receptor_file_list,dtype=tf.string)

filename_queue = tf.train.slice_input_producer([ligand_files,receptor_files],num_epochs=10,shuffle=True)

serialized_ligand = tf.read_file(filename_queue[0])
serialized_receptor = tf.read_file(filename_queue[1])

init_two = tf.initialize_all_variables()
init_one = tf.initialize_local_variables()
sess = tf.Session()
coord = tf.train.Coordinator()
threads = tf.train.start_queue_runners(sess=sess,coord=coord)
sess.run([init_one])
sess.run([init_two])

while True:
    print sess.run([serialized_ligand,serialized_receptor])

Tags: run数据库queueinittftrainfilesfilename

热门问题