pytorch/benchmark

Pull Requests Commits

Refactor Cyclegan

wconstab committed 4 years ago

1a727590

Address tacotron and CycleGan issues with niter

wconstab committed 4 years ago

d2a08994

Fix niter and lazy device issues

wconstab committed 4 years ago

450d2716

fix niter kwarg for moco, stargan, cyclegan

wconstab committed 4 years ago

9092cd41

Add TFLOPS measure for resnet50 model (#668)

xuzhao9 committed 4 years ago

5c93a538

Fix assert on tuple (#692)

kit1980 committed 4 years ago

26ff64fe

Switch to python 3.8 as default in CI and readme. (#686)

xuzhao9 committed 4 years ago

5d5f4d71

Enable V2 workflow (#671)

xuzhao9 committed 4 years ago

dd8d7c98

Update opacus_cifar10 to latest Opacus API (#681)

zou3519 committed 4 years ago

6213f072

Don't change behavior based on apex package in yolov3 (#677)

jansel committed 4 years ago

f84e9458

Remove prints while running LearningToPaint (#674)

jansel committed 4 years ago

7a95f5bd

Remove printouts from running benchmarks (#676)

jansel committed 4 years ago

0c920ebb

Don't use jit when jit=False for pyhpc benchmarks (#675)

jansel committed 4 years ago

e2aa07a8

Fix broken main branch. Disable half when using CPU device. (#667)

xuzhao9 committed 4 years ago

e141d183

Add fp16 and fx2trt support to all torchvision models (#663)

xuzhao9 committed 4 years ago

Verified ea08835c

Add fx2trt support to resnet50 (#661)

xuzhao9 committed 4 years ago

3416dddd

Improve the quality of moco train (#659)

aaronenyeshi committed 4 years ago

edc0ab7b

Improve the quality of speech transformer (#656)

xuzhao9 committed 4 years ago

0d0fb42f

Improves the model quality of huggingface T5 (#644)

xuzhao9 committed 4 years ago

b00581f6

Fix size of equation_of_state and isoneutral_mixing, add tke (#651)

xuzhao9 committed 4 years ago

e366637a

Improve the quality of quantized_qat models. (#640)

xuzhao9 committed 4 years ago

f71e3c42

Improve quality of pytorch_stargan (#648)

xuzhao9 committed 4 years ago

3970f85b

Improve the quality of demucs model (#639)

xuzhao9 committed 4 years ago

a53ff27d

Improve the quality of soft actor critic model (#645)

xuzhao9 committed 4 years ago

bf3c5d49

Remove torch requirements in requirements.txt (#653)

xuzhao9 committed 4 years ago

70d09e8d

Improve the quality of tts_angular, fix train and eval batch sizes (#634)

xuzhao9 committed 4 years ago

f41a4a08

Make the script more robust on supporting unknown devices (#642)

Jiewen Tan committed 4 years ago

79798c58

Make the model robust to accept Lazy device (#646)

Jiewen Tan committed 4 years ago

c2b7cacc

Unconditionally move inputs to the specified device (#637)

Jiewen Tan committed 4 years ago

8f5048a4

Fix yolov3 model training to original batch size (#630)

xuzhao9 committed 4 years ago

0954aa61

Older