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

Question

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

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 |
+--------------+----------+-------+------+-----+

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

Использование Python 3.7.3 и Ray 0.8.4

Answer 1

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