以下是一些CentOS上PyTorch的可视化工具推荐:
pip install hiddenlayer
,可通过build_graph
方法生成网络结构图并保存为图片。pip install torchviz
,使用make_dot
函数生成可视化图。pip install tensorboard
,需在代码中使用SummaryWriter
记录数据,启动后可在浏览器查看。pip install tensorboardX
,用法与TensorBoard类似。pip install netron
,通过netron.show
方法展示模型。