摘要:
pytorch 模型参数的保存与加载 pytorch 保存与加载模型参数的最主要的三个函数 torch.save:将序列化对象保存到磁盘。此函数使用Python的pickle模块进行序列化。使用此函数可以保存如模型、tensor、字典等各种对象。 torch.load:使用pickle的unpick 阅读全文
摘要:
pytorch模型转换为onnx,并使用netron可视化 netron 是一个非常好用的网络结构可视化工具。 但是netron对pytorch模型的支持还不成熟。自己试的效果是生成的模型图没有连线。 目前支持的框架 根据netron的github 目前netron支持: ONNX (.onnx, 阅读全文
摘要:
深度学习框之 静态图&动态图 各位炼丹者应该都会有自己常用的一种或几种深度学习框架,如 MxNet、Caffe、Tensorflow、Pytorch、PaddlePaddle(百度),甚至是国产新兴框架 MegEngine(旷视)、MindSpore(华为)等,在涉及介绍这些框架的时候,都会提及动态 阅读全文
摘要:
衡量AI算力的“FLOPS” 什么是FLOPS FLOPS,是“每秒所执行的浮点运算次数”(floating-point operations per second) 的缩写。它常被用来估算电脑的执行效能,尤其是在使用到大量浮点运算的科学计算领域中。 在这里所谓的“浮点运算”,实际上含括了所有涉及小 阅读全文