Как визуализировать мою историю тренировок в pytorch?

Question

Как вы, ребята, представляете историю тренировок вашей модели pytorch, как в керасе здесь .

У меня есть обученная модель с pytorch, и я хочу увидеть график ее обучения.Могу ли я сделать это, используя только matplotlib?Если да, может кто-нибудь дать мне ресурсы для подражания.

Answer 1

Вы должны сохранить потерю во время тренировки.У обученной модели не будет истории ее потери.Вам нужно тренироваться снова.

Сохраните потерю во время тренировки, затем нанесите на карту ее время, используя matplotlib.В вашей функции обучения, где рассчитывается потеря, сохраните ее в файл и визуализируйте позже.Кроме того, вы можете использовать tenorboardX, если хотите визуализировать в режиме реального времени.

Это руководство для тензорной доски X: http://www.erogol.com/use-tensorboard-pytorch/