pytorch/benchmark

Pull Requests Commits

Fix bug and disable tests that fail CI

aaronenyeshi committed 4 years ago

e26be486

densenet121: remove infer examples as 1st element

aaronenyeshi committed 4 years ago

af7e9735

Fix the train batch size of densenet121

aaronenyeshi committed 4 years ago

e660c1ce

Checkout the lazy_tensor_staging branch from the pytorch repo. (#615)

xuzhao9 committed 4 years ago

9f4b27c3

Remove pathlib install command as it is built-in 3.8. (#613)

xuzhao9 committed 4 years ago

ef847774

Use the correct batch size for vgg16 training (#583)

xuzhao9 committed 4 years ago

7eb31b1d

Added lazy tensor testing CI. (#600)

xuzhao9 committed 4 years ago

bb7256f2

Fixes squeezenet hierarchical batching. (#609)

xuzhao9 committed 4 years ago

b0a71ba7

Set the correct train batch size for squeezenet. (#574)

xuzhao9 committed 4 years ago

cee0ce46

Fix the runner temp interface. (#607)

xuzhao9 committed 4 years ago

c2e36b87

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

Older