深度学习工程师秘而不宣的代码重构三原则

　　去年春晚，我蹲在央视云上那个训练集群的后台看日志，不是调参，是救火——模型上线前两小时，inference pipeline崩了三次，每次都是同一个函数，它不报错，就卡在 torch.jit.script 那行，像人突然失忆，站在电梯口反复按开门键。

　　李工端着保温杯过来，拧开盖子吹了三下，说“重写吧”，我说“你先告诉我怎么重”，他低头戳手机，微信弹出一条：「别动结构，只换 kernel」。我没回，把终端最小化，切到浏览器，开了三十个标签——十七个是 GitHub PR，五个是 PyTorch nightly 的 changelog，剩下七个全是自己昨天写的 debug note，标题都带“？？？”。

　　第一原则：删掉所有自以为“将来会用”的分支逻辑。不是删 if，是删整个 if 里裹着的那层 try-catch-finally-else-if-logging-wrapping-wrapper。你真信那个注释里写的“支持 future quantization”？我原以为那是接口兼容性设计，结果翻 commit 记录发现，那行注释是实习生提 PR 时随手敲的，他三天后转岗去做 OCR 数据清洗了。这个我敢打包票。

　　第二原则：把 model.forward() 拆成至少两个独立函数，一个纯 tensor 流水线，一个只管 shape 校验和 device 拓扑对齐。但别叫它们 validate_input 和 run_core——就叫 core 和 align。名字越糙，越没人敢往里塞 logging 或 metrics 上报。上周王磊硬在 align 里加了一行 wandb.log，第二天 infer 延迟从三百七十二毫秒跳到八百多，没报警，监控大屏一直在闪红光，没人理，因为告警阈值设的是秒级。

文章配图，仅供参考

　　窗外天黑了，但机房灯一直没开，靠服务器散热风扇的蓝光映着玻璃。我盯着屏幕上那个被划掉又恢复的 @torch.compile 装饰器，突然想起前天吃饭时听见隔壁桌两个实习生聊：“重构就是让代码看起来更像论文”。我笑出声，对面运维老张抬头看了我一眼，筷子停在半空，夹着一块凉透的藕片，没说话。

　　优胜劣汰不是说跑得快的活下来——是那些懒得加 telemetry、拒绝接 tracing context、把 config.yaml 当 throwaway 的模型，最后反而撑过了三轮架构升级。不是它们更强，是它们太“脏”，没人敢动，动了怕崩；太“哑”，崩了也没日志；太“旧”，连文档都丢了，只能靠 grep 找依赖。这种代码活得久，不是因为它好，是因为它没用——没用的东西，才没人改。

　　我原以为“秘而不宣”是怕别人学走绝招，结果跟陈博士在食堂吵架才知道，他们根本不藏，是压根没意识到这算原则——就像厨师不会告诉你“炒菜时别用水洗热锅”，因为那不是技巧，是常识崩塌后的残渣。他当时正往酸梅汤里加冰块，冰块掉进不锈钢盆里“叮”一声，很脆。那声音让我愣住三秒，手里的勺子悬在半空。

　　桌上的外卖盒还没拆封，酱汁已经渗到纸盒边缘，洇成深褐色一圈。监控群里弹出新消息：「infer-qps 下跌 12%（基准毛估）」。没人问为什么，只刷了一排“+1”，还有人发了个咖啡表情。我把那行被划掉的 @torch.compile 又点回去，按 Ctrl+S，文件名自动变成 model_v3_refactored_broken_but_faster.py——后缀是自动加的，IDE 插件，叫 “TorchLint-Sanity”。

　　现在我遇到要重构的模型，第一件事不是拉 branch，是先搜项目里所有 os.environ.get('DEBUG_')。有，就删；没，就加一行 os.environ['NO_TORCH_JIT_FALLBACK'] = '1'，然后直接 push。推完立刻去楼下便利店买烟，不是抽，是拿打火机烤 SD 卡背面的芯片，等它烫手了再插回训练节点——纯属心理安慰，但比坐那儿等 CI 快。

　　所以别问我“怎么判断该不该重构”。你看到 model.state_dict().keys() 里混着 ‘module.’ 和 ‘_orig_mod.’，看到 loss.backward() 后还跑 optimizer.step() 两次，看到 trainer.fit() 里手动 del 模型——这时候不是原则在起作用，是你指尖发麻，呼吸变浅，脑子里只剩一个念头：这坨东西，必须死一次。

　　下周我约了李工对线，主题是「要不要把 onnxruntime-gpu 降级到 1.16.3」。他坚持要降，说新版 async IO 有内存泄露，但我怀疑是他的 custom op 没适配。我们约在B座负二层茶水间，他带保温杯，我带充电宝。充电宝电量显示 87%，但插上后纹丝不动——可能是 USB-C 线芯断了两根。这事跟重构没关系，但我想起来了，就写了。

（编辑：PHP编程网 - 金华站长网）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!