pytorch/benchmark

Pull Requests Commits

Add pygame with gym requirement.

kit1980 committed 3 years ago

e9643827

Enable fp16 for torchvision models inference CUDA tests by default (#758)

xuzhao9 committed 3 years ago

2ee0fd06

Fix a few problems discovered when running the workflow. (#756)

xuzhao9 committed 3 years ago

d6686732

Enable torch_trt and handle subprocess segfault in subprocess_worker (#754)

xuzhao9 committed 3 years ago

0c9e3002

Revert #736 with maml and maml_omiglot (#752)

xuzhao9 committed 3 years ago

14d06881

Add nvfuser support for torchvision models (#744)

xuzhao9 committed 3 years ago

775b254a

Enable fx2trt and torch_tensorrt support across all TB models (#743)

xuzhao9 committed 3 years ago

e0deda39

Adds the sweep runner script (#742)

xuzhao9 committed 3 years ago

69723a96

Move fx2trt out of PyTorch core (#72499)

Yinghai Lu committed 3 years ago

f91d1939

Remove the maml models as they have poor quality and should be replaced. (#736)

xuzhao9 committed 3 years ago

907a4eba

Add a new "test" argument to models (#735)

xuzhao9 committed 3 years ago

b93c9ebc

Support torch_tensorrt for torchvision and timm models (#732)

xuzhao9 committed 3 years ago

6cf0ec0a

changed eval_bs to train_bs (#731)

erichan1 committed 3 years ago

453b22eb

Silence printouts in fastNLP_Bert and vision_maskrcnn (#722)

jansel committed 3 years ago

a389101d

Add the test workflow. (#729)

xuzhao9 committed 3 years ago

67a68c9c

Fix a bug when there is no anomaly detected (#728)

xuzhao9 committed 3 years ago

0f239843

Use py38 and CUDA 11.3 as default in the nightly wheel script (#718)

xuzhao9 committed 3 years ago

88125e01

Fix the installer script path (#721)

xuzhao9 committed 3 years ago

dc42262b

Allow running install.py from other directories (#719)

jansel committed 3 years ago

830e1654

Remove the unused waveglow requirements file (#716)

xuzhao9 committed 3 years ago

76fa3723

Fixed the workflow file and upload script. (#712)

xuzhao9 committed 3 years ago

aca0584e

Fix V2 bisection workflow bug. (#709)

xuzhao9 committed 3 years ago

5f707392

Enable CUDA graph train on torchvision models (#706)

xuzhao9 committed 3 years ago

e50b595b

Workaround the nightly package hash issue. (#705)

xuzhao9 committed 3 years ago

a2070588

Officially enable the v2 workflow, and upload the results to scribe (#704)

xuzhao9 committed 3 years ago

9083e7a5

Fix issues encountered when running models with lazy tensors (#697)

wconstab committed 3 years ago

bf51bcff

Replace distutils strtobool in gen-summary-md (#703)

aaronenyeshi committed 3 years ago

854525f2

Cleanup devbig related code. (#702)

xuzhao9 committed 3 years ago

ad3ba488

Remove the use of distro.linux_distribution() (#701)

xuzhao9 committed 3 years ago

559e7298

Add TorchBench V2 score (#700)

xuzhao9 committed 3 years ago

e6d216b1

Older