Comments 6
название "Визуализация модели PyTorch" звучит неоднозначно. Более корректно было бы "Визуализация архитектуры модели PyTorch"
Короче графическое представление модели PyTorch может быть затруднительным из-за того, что PyTorch использует динамический граф вычислений в отличие от статического графа, который используется в других библиотеках для глубокого обучения. Динамический граф означает, что граф строится по мере выполнения кода, что делает его более гибким и удобным для работы, но усложняет его визуализацию. А еще там циклы и ветвления...
А как же Tensor Board ? или Graph Viz ?
Это как вы `Graph Viz` в чистом виде примените? Да, pytorchviz его использует как бэкенд.
Всё так, сам использую Netron как внешний просмотровщик, и самостоятельно допиленную версию pytorchviz внутри iноутбуков. Очень жаль, что Нетрон на JavaScript-е написан.
Вопрос, почему так сложно визуализовать торч-модель?
В Tensorflow аналогичный этому примеру FunctionalAPI визуализируется без проблем
через keras.utils.plot_model()
Визуализация архитектуры модели PyTorch