#matplotlib #pytorch #graph-visualization
#matplotlib #pytorch #график-визуализация
Вопрос:
Как вы, ребята, визуализируете историю обучения вашей модели pytorch, как в keras здесь.
У меня есть обученная модель pytorch, и я хочу увидеть график ее обучения. Могу ли я сделать это, используя только matplotlib? Если да, может ли кто-нибудь предоставить мне ресурсы для отслеживания.
Ответ №1:
Вы должны сохранить потери во время тренировки. У обученной модели не будет истории ее потерь. Вам нужно снова тренироваться.
Сохраните потери во время обучения, затем сопоставьте их с эпохами, используя matplotlib. В вашей обучающей функции, где вычисляются потери, сохраните это в файл и визуализируйте позже. Кроме того, вы можете использовать tensorboardX, если хотите визуализировать в реальном времени.
Это руководство по tensorboardX: http://www.erogol.com/use-tensorboard-pytorch /
Комментарии:
1. Как я могу сохранить потерю? поместить это в новую переменную? Извините, я действительно не знаю.
2. dpaste.com/0T9A4GH вот код. Изначально я получил код из github.com/paraschopra/one-network-many-uses . Надеюсь, вы сможете мне помочь 🙂
3. в строке 49. запишите эту потерю в файл.
4. в текстовый файл или ввести переменную?
5. как я уже сказал, для текстового файла