摘要: 深度学习中网络的加速主要有如下的几种方式: 1)设计高效且小的网络,如MobileNet系列、shuffleNet系列、VoVNet等;2)从大的模型开始通过量化、剪裁、蒸馏等压缩技术实现网络的小型化;3)在inference阶段使用特殊的计算库实现计算的加速,比如MKL、TensorRT等;在上面 阅读全文
posted @ 2021-08-03 11:58 鲍曼小学生 阅读(687) 评论(0) 推荐(0) 编辑
摘要: 阅读全文
posted @ 2021-08-03 11:06 鲍曼小学生 阅读(35) 评论(0) 推荐(0) 编辑
摘要: 在pytorch进行模型保存的时候,一般有两种保存方式,一种是保存整个模型,另一种是只保存模型的参数。 torch.save(model.state_dict(), "my_model.pth") # 只保存模型的参数 torch.save(model, "my_model.pth") # 保存整个 阅读全文
posted @ 2021-08-03 09:43 鲍曼小学生 阅读(1540) 评论(0) 推荐(0) 编辑