TorchBench: Benchmarking PyTorch with High API Surface Coverage

最近,我们在arXiv发布了TorchBench的第一篇论文,TorchBench: Benchmarking PyTorch with High API Surface Coverage。代码开源在Github TorchBench TorchBench是一个基于PyTorch的深度学习框架的benchmarking框架。在这篇论文中,我们介绍了TorchBench目前包含的深度 more ...