Ни один из примеров в документации не показывает, как ведение журнала происходит с классом API? Вы все еще используете tune.track.log даже в классе API?
tune.track.log
Запись в тензор потока должна происходить автоматически при использовании API класса - я обновлю документы, чтобы сделать это более понятным.