pytorch模型可視化

先把tensorflow和pytorch安裝好 之後在pytorch環境裏 pip install tensorboardX 將 with SummaryWriter(comment=‘LeNet’) as w: w.add_graph(model, (varInput,)) 添加到模型和輸入的後面，類似這樣注意參數第一個是模型，第二個是輸入 然後將數據集和測試集變小一點，迭代次數設爲一代，跑一次

>>阅读原文<<