1. 程式人生 > >pytorch模型視覺化2:tensorboardX

pytorch模型視覺化2:tensorboardX

不同於tensorflow提供了強大的視覺化功能,pytorch視覺化一般藉助第三方工具,更通用的方法是使用tensorboardX。
https://github.com/lanpa/tensorboardX

使用命令:pip install tensorboardX安裝後,如下程式碼視覺化模型:

print(model)
writer = SummaryWriter(log_dir="./logs/",comment="myresnet")
with writer:
    writer.add_graph(model,input_to_model=torch.rand(1,3,32
,32))

在這裡插入圖片描述
更多內容,訪問文件:
https://tensorboardx.readthedocs.io/en/latest/index.html