跳到正文
原文
Reddit · r/MachineLearning· /u/AKJ7·· 2 小时前AI 评分18

pytest-torch:用于简化 PyTorch 模型测试的开源 pytest 插件发布

Introducing pytest-torch [P]

AI 导读

开发者 AKJ7 发布开源 pytest 插件 pytest-torch,用于简化 PyTorch 模型测试。插件内置推理检查、compile/export 一致性测试、checkpoint 测试、精度评估以及逐层诊断,可定位两个模型产生分歧的位置。该项目目前处于 alpha 阶段,作者欢迎社区反馈和 star。

来源:Reddit · r/MachineLearning · reddit.com