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