Pytorch可視化

一，利用 tensorboardX 可視化網絡結構 參考 https://github.com/lanpa/tensorboardX 支持scalar, image, figure, histogram, audio, text, graph, onnx_graph, embedding, pr_curve and video summaries. 例子要求tensorboardX>=1.2 an

>>阅读原文<<