pytorch tensorboard模型結構顯示

轉自：https://www.cnblogs.com/jfdwd/p/11445135.html 重點關注最後兩句，其中使用了python的上下文管理，with 語句，可以避免因w.close未寫造成的問題。推薦使用此方式。 使用 w.add_graph(model, (dummy_input,))，其中第一個參數爲需要保存的模型，第二個參數爲輸入值，tuple類型。打開tensorboard控制

>>阅读原文<<