pytorch/benchmark

Pull Requests Commits

Support skipping tests on a blacklist in test.py (#584)

aaronenyeshi committed 4 years ago

75c0da2e

Only load the module specified in run.py argument (#577)

xuzhao9 committed 4 years ago

1941a66f

Use the original batch size from mobilenet_v2 paper. (#576)

xuzhao9 committed 4 years ago

a8023c67

Fix the train batch size of alexnet. (#575)

xuzhao9 committed 4 years ago

b4c316b5

Adding maskrcnn model from torchvision (#547)

xuzhao9 committed 4 years ago

39b5b6d3

Fix the fastNLP model train batch size. (#571)

xuzhao9 committed 4 years ago

0d1da2b8

Fix the jit code for timm models. (#572)

xuzhao9 committed 4 years ago

dc342679

Correct LearningToPaint to match the wild (#560)

aaronenyeshi committed 4 years ago

17cd2f96

Add time_lazy.py script for lazy tensor measurement

wconstab committed 4 years ago

4fbf86a1

Make bert_pytorch support lazy device

wconstab committed 4 years ago

6f602558

Add check_lazy.py to sweep models and collect stats

wconstab committed 4 years ago

31e733d3

Fix attention_is_all_you_need_pytorch model (#562)

xuzhao9 committed 4 years ago

1b71e91b

Add detectron2 maskrcnn model (#556)

xuzhao9 committed 4 years ago

12442efc

Temporarily remove the attention model and fix pytorch_struct model. (#558)

xuzhao9 committed 4 years ago

e71b8d04

Fix GPU utilization of squeezenet1_1 (#552)

xuzhao9 committed 4 years ago

b9136736

Fix GPU utilization issue of alexnet model (#555)

xuzhao9 committed 4 years ago

04e0ac05

Fix GPU utilization issue of mnasnet1_0 model (#548)

xuzhao9 committed 4 years ago

626f1d71

Fix GPU utilization issue of vgg16 (#553)

xuzhao9 committed 4 years ago

8b54c005

Fix GPU utilization issue for shufflenet_v2 (#551)

xuzhao9 committed 4 years ago

477266b4

Fix GPU utilization issue for resnet18 (#549)

xuzhao9 committed 4 years ago

cd9a7f75

Fix GPU utilization issue of resnext50_32x4d model (#550)

xuzhao9 committed 4 years ago

4af5f314

Fix get_module() API for 9 benchmarks (#479)

jansel committed 4 years ago

413c3bda

Fix mobilenet_v2 GPU utilization. (#544)

xuzhao9 committed 4 years ago

ba3b89c3

Add a ondemand CI to test GPU stability (#538)

xuzhao9 committed 4 years ago

626d6143

Delete models that I couldn't install (NOT FOR MERGE)

wconstab committed 4 years ago

fb1ff187

[LTC][Not for merge] Enabling Lazy Tensor in hf_Bert and resnet50

Jiewen Tan committed 4 years ago

65a44fb6

Fix the import issue of speech_transformer. (#542)

xuzhao9 committed 4 years ago

bdea82c6

Fix GPU utilization issue of mobilenet_v3_large (#543)

xuzhao9 committed 4 years ago

1b2fd45c

Fix GPU utilization issue of timm_vovnet and timm_vision_transformer (#540)

xuzhao9 committed 4 years ago

e6ef04b4

Fixes regnet and resnest models (#535)

xuzhao9 committed 4 years ago

107cd55d

Newer Older