Как получить два скаляра на одном графике с tenorboardX? - PullRequest
Новая
       77

Как получить два скаляра на одном графике с tenorboardX?

0 голосов
/ 02 декабря 2018

Документы, кажется, указывают, что add_custom_scalars_multilinechart делает это, но это не работает.

Есть что-то вроде этого: 

from tensorboardX import SummaryWriter
writer = SummaryWriter(comment='test')

writer.add_custom_scalars_multilinechart(['loss/train', 'loss/test'], title='losses')

for blahblah:
    ...
    writer.add_scalar('loss/train', loss.item(), epoch)
    writer.add_scalar('loss/test', loss_test.item(), epoch)

1 Ответ

0 голосов
/ 15 января 2019

Постройте два скаляра на одном графике с tenorboardX: 

from tensorboardX import SummaryWriter

Создайте два summaryWriter для двух скаляров 

writer_train = SummaryWriter('runs/train_0')
writer_test = SummaryWriter('runs/test_0')

Добавьте экземпляры скаляров в summaryWriter соответственно;они должны иметь одинаковый тег, например: "LOSS" 

for data in loop:
    writer_train.add_scalar('LOSS', loss.data.item(), idx)
    writer_test.add_scalar('LOSS', loss_test.data.item(), idx)

Для рабочего кода, пожалуйста, посетите github: Примеры с tenorboardX (см. more_plots_one_chat.py)

Учебное пособие: ТензорбордX

...