我正在使用tf.data.TFRecordDataset從TFRecord文件讀取數據集。在

我想知道每一步都在處理哪個時代。在

詳細信息如下：100個樣本保存在一個TFRecord文件中，batch_size設置為50，epoch_num設置為5。在

下面是我的簡化代碼：def read_and_decode_TFRecordDataset(tfrecords_path, batch_size, epoch_num):

dataset = tf.data.TFRecordDataset(tfrecords_path)

dataset = dataset.map(parser_deblur)

epoch = tf.data.Dataset.range(epoch_num)

dataset = epoch.flat_map(lambda i: tf.data.Dataset.zip(

(dataset, tf.data.Dataset.from_tensors(i).repeat())))

dataset = dataset.repeat(epoch_num).shuffle(1000).batch(batch_size)

iterator = dataset.make_one_shot_iterator()

(face_blur_batch, face_gt_batch), epochNow = iterator.get_next()

return face_blur_batch, face_gt_batch, epochNow

print EPOCH: {epochNow}, STEP: {step}

我期望的是：

^{pr2}$

但實際產出是：

^{3}$

我不知道什么是紀元？這似乎是隨機的。每次跑步都不一樣。在

你知道如何修復上面的代碼嗎？或者如何通過其他方法獲得紀元計數器？在

總結

以上是生活随笔為你收集整理的epoch如何设置_使用TFRecordDataset时如何设置epoch计数器？的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。