pytorch/benchmark

Pull Requests Commits

Cleanup the benchmarking code for timm models (#772)

xuzhao9 committed 3 years ago

f56637b2

Add the actual TensorRT correctness testing code (#763)

xuzhao9 committed 3 years ago

b904ffdb

First PR to add the correctness checking code to eval tests (#762)

xuzhao9 committed 3 years ago

ca731c26

Move the jit script code to __post_init__ (#761)

xuzhao9 committed 3 years ago

3173d623

Add pygame with gym requirement (#766)

kit1980 committed 3 years ago

0743d73f

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 4 years ago

0c9e3002

Revert #736 with maml and maml_omiglot (#752)

xuzhao9 committed 4 years ago

14d06881

Add nvfuser support for torchvision models (#744)

xuzhao9 committed 4 years ago

775b254a

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

xuzhao9 committed 4 years ago

e0deda39

Adds the sweep runner script (#742)

xuzhao9 committed 4 years ago

69723a96

Move fx2trt out of PyTorch core (#72499)

Yinghai Lu committed 4 years ago

f91d1939

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

xuzhao9 committed 4 years ago

907a4eba

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

xuzhao9 committed 4 years ago

b93c9ebc

Support torch_tensorrt for torchvision and timm models (#732)

xuzhao9 committed 4 years ago

6cf0ec0a

changed eval_bs to train_bs (#731)

erichan1 committed 4 years ago

453b22eb

Silence printouts in fastNLP_Bert and vision_maskrcnn (#722)

jansel committed 4 years ago

a389101d

Add the test workflow. (#729)

xuzhao9 committed 4 years ago

67a68c9c

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

xuzhao9 committed 4 years ago

0f239843

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

xuzhao9 committed 4 years ago

88125e01

Fix the installer script path (#721)

xuzhao9 committed 4 years ago

dc42262b

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

jansel committed 4 years ago

830e1654

Remove the unused waveglow requirements file (#716)

xuzhao9 committed 4 years ago

76fa3723

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

xuzhao9 committed 4 years ago

aca0584e

Fix V2 bisection workflow bug. (#709)

xuzhao9 committed 4 years ago

5f707392

Enable CUDA graph train on torchvision models (#706)

xuzhao9 committed 4 years ago

e50b595b

Workaround the nightly package hash issue. (#705)

xuzhao9 committed 4 years ago

a2070588

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

xuzhao9 committed 4 years ago

9083e7a5

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

wconstab committed 4 years ago

bf51bcff

Newer Older