导读
动手对照之前,先讲清楚两件事:这一卷的数字怎么来,以及值不值得转。
- 怎么读这一卷:锁定的版本和 burn 的 feature;每章的结构;一对对照程序怎么跑、怎么用
compare.py逐个数比较;为什么 f32 下两边的结果不逐位相同,判断用绝对误差、容差取 1e-6 的依据;三条主线每一步落在哪一页。 - 为什么要转:同一个 MLP 推理,Python 加 torch 约 729 MiB、启动约 0.65 秒、峰值 RSS 约 208 MiB,burn 是一个 2.3 MiB 的二进制、启动约 3 毫秒、峰值 RSS 约 2.6 MiB;也写了不值得转的情况:
unique、searchsorted、bincount、linalg.inv在 burn 0.22.1 里找不到,svd不能求导。
flowchart TB read[怎么读这一卷] --> ver[锁定版本] read --> run["对照怎么跑"] read --> tol["对得上的标准"] read --> lines[三条主线] why[为什么要转] --> size["体积 启动 内存"] why --> cov[算子覆盖] why --> notworth[不值得转的情况]
💬 关联讨论
发起讨论加载中...