pytorch模型可视化

先把tensorflow和pytorch安装好 之后在pytorch环境里 pip install tensorboardX 将 with SummaryWriter(comment=‘LeNet’) as w: w.add_graph(model, (varInput,)) 添加到模型和输入的后面，类似这样注意参数第一个是模型，第二个是输入 然后将数据集和测试集变小一点，迭代次数设为一代，跑一次

>>阅读原文<<