RuntimeError («Глобальный шаг должен быть создан для использования ProfilerHook.») При назначении хуков для tf.contrib.training.evaluate_repeatedly ()

Question

Я воспроизводлю DeepLab hnas_nasnet и профилирую модель. Я хочу профилировать производительность вывода (время и память) путем создания файла timeline.json, но выпущенный код в / model / research / deeplab не написан ни slim, ни оценщиком. Поэтому я отредактировал код: hooks = [summary_hook] чтобы: hooks = [tf.train.ProfilerHook (save_steps = 1, output_dir = "profile /")]

Я пытался добавить tf_global_step = slim.get_or_create_global_step (), но это не сработало.

hooks = [tf.train.ProfilerHook(save_steps=1, output_dir="profile/")]
tf.contrib.training.evaluate_repeatedly(
        master=FLAGS.master,
        checkpoint_dir=FLAGS.checkpoint_dir,
        eval_ops=[update_op],
        max_number_of_evaluations=num_eval_iters,
        hooks=hooks,
        eval_interval_secs=FLAGS.eval_interval_secs)

Если это не помогло, есть ли другие рекомендуемые способы для профилирования в этом конкретном случае?