tf.Session (). run () выдает ошибку итератора тензорного потока

Question

Я пытаюсь написать версию Google-Bert tenorflow1 без использования tf.Estimator, так как в нем меньше гибкости для добавления пользовательских слоев. я получаю следующую ошибку:

tensorflow.python.framework.errors_impl.NotFoundError: Resource AnonymousIterator/AnonymousIterator0/N10tensorflow4data16IteratorResourceE does not exist.
         [[{{node IteratorGetNext}}]]

получаю эту ошибку, когда

набор данных извлекается из tf.record файла.

способ, которым вызывается function_function:

with tf.Session() as sess:
    for raw_record in d.take(1):
            print( raw_record) # is dictionary with following keys-input_id, input_mask...etc 
            print(type(raw_record)) # dict
            a = model_fn(raw_record)
            init = tf.initialize_all_variables()
            sess.run(init)
            sess.run(a) # throws error
            print(a)