Как я могу использовать tqdm для визуализации прогресса этапов обучения с использованием API tf.data.Dataset?

Question

Я хочу использовать tqdm для визуализации шагов обучения по сети cnn.
Как я могу реализовать tqdm с tf.data.Dataset() api?
Можете ли вы показать мне пример кода?ТНХ!

Answer 1

Это довольно просто:

1. Получите количество выборок в вашем наборе данных,

2. Затем преобразуйте число в некоторую итеративную структуру Python.

   for _ in tqdm(iterable=xxx, total=num_samples):
        batch_data = sess.run(ele_derived_from_tf_dataset)