Добавление дополнительных метрик в таблицу состояния Ray Tune (Python, PyTorch)

Question

При запуске tune.run() на наборе конфигов для поиска, возможно ли добавить дополнительные столбцы метрик (например, a, b, и т. Д. c) в распечатанную таблицу состояния?

tune.track.log(a=metric1, b=metric2)

даст следующую таблицу без столбцов для метрик a и b:

== Status ==
Memory usage on this node: 22.1/125.8 GiB
Using FIFO scheduling algorithm.
Resources requested: 1/32 CPUs, 1/4 GPUs, 0.0/65.59 GiB heap, 0.0/22.13 GiB objects
Result logdir: /home/nyxynyx/ray_results/fooba
Number of trials: 4 (3 PENDING, 1 RUNNING)
+--------------+----------+-------+------+-----+
| Trial name   | status   | loc   |   lr |   x |
|--------------+----------+-------+------+-----|
| fooba_00000  | RUNNING  |       | 0.01 |   1 |
| fooba_00001  | PENDING  |       | 0.1  |   1 |
| fooba_00002  | PENDING  |       | 0.01 |   5 |
| fooba_00003  | PENDING  |       | 0.1  |   5 |
+--------------+----------+-------+------+-----+

Как мы можем включить столбец для каждого показателя c, который мы передаем tune.track.log() кроме mean_accuracy?

Использование Python 3.7.3 и Луч 0.8.4

Answer 1

Да! Вы должны быть в состоянии сделать это с объектом репортера: https://ray.readthedocs.io/en/latest/tune/api_docs/reporters.html