pytest-torch：面向 PyTorch 模型测试的开源 pytest 插件

Introducing pytest-torch [P]

AI 导读

开源 pytest 插件 pytest-torch 发布，用于简化 PyTorch 模型测试，目前处于 alpha 阶段。它提供推理检查、compile/export 一致性验证、checkpoint 测试、精度评估和逐层诊断，可定位两个模型出现分歧的位置。作者正征集反馈，尤其是难以测试或复现的模型 bug 案例。

正文

I built pytest-torch to make testing PyTorch models easier

Writing the same device setup, reproducibility checks, and output comparisons across projects gets repetitive, so I built pytest-torch, an open-source pytest plugin.

It includes inference checks, compile/export parity, checkpoint tests, accuracy evaluation, and layer-by-layer diagnostics to help locate where two models diverge.

It’s currently in alpha, and I’d appreciate feedback—especially examples of model bugs that are frustrating to test or reproduce.

GitHub: https://github.com/AKJ7/pytest-torch

Thanks and please drop a star if possible.

submitted by /u/AKJ7
[link] [留言]