pytorch tensorboard模型結構顯示

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