pytorch/benchmark

Pull Requests Commits

Remove V0 workflow. Use CUDA 11.3 for V1 workflow. (#596)

xuzhao9 committed 4 years ago

26229b6e

Use opacus version < 1.0 to mitigate BC-breaking API change (#605)

xuzhao9 committed 4 years ago

3e7bf736

Update the train_bs and eval_bs for super_slomo. (#595)

xuzhao9 committed 4 years ago

1dbe4bc7

Improve test.py EXCLUDELIST to disable by model, test, and device (#592)

aaronenyeshi committed 4 years ago

5e898020

Fix train batch size for attention model. (#593)

xuzhao9 committed 4 years ago

488d2071

Fix train batch size for pytorch_struct. (#591)

xuzhao9 committed 4 years ago

121ea023

Fix background matting batch size. (#594)

xuzhao9 committed 4 years ago

5b9b32f9

Fix the train options of LearningToPaint (#581)

aaronenyeshi committed 4 years ago

a5213f1c

Fix the train arch of DLRM (#580)

aaronenyeshi committed 4 years ago

149410d8

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

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

Newer Older