花好多天时间千辛万苦才找到的代码,却复现不出来,谁能 救救我?不要着急,这个被称为深度学习代码救星的github项目绝对可以帮到你,目前在github上星标已经 36.8k 了。这是一个对深度学习前沿架构、算法进行逐行解读的可视化项目。逐行解读对初学者非常友好。
使用工具如 Matplotlib 来可视化数据分布,以及使用 Graphviz 或 TensorBoard 来可视化模型结构。在 PyTorch 中,可以这样保存和加载模型:九、使用 GPU 加速训练。