摘要: (本文首发于公众号,没事来逛逛) Pytorch1.8 发布后,官方推出一个 torch.fx 的工具包,可以动态地对 forward 流程进行跟踪,并构建出模型的图结构。这个新特性能带来什么功能呢?别的不说,就模型量化这一块,炼丹师们有福了。 其实早在三年前 pytorch1.3 发布的时候,官方 阅读全文
posted @ 2022-03-17 09:49 大白话AI 阅读(1773) 评论(0) 推荐(0) 编辑