NVIDIA/Megatron-LM

[QUESTION] How to Visualize Computational Graph

Opened this issue · 0 comments

zixianwang2022 commented

Your question
Hi,

Is there a way I can visualize the computational graph during backward propagation in Megatron?