pytorch/benchmark

Pull Requests Commits

Add model torchrec dlrm (#1397)

xuzhao9 committed 3 years ago

bd90678d

check for fx2trt only for cuda device (#1395)

snadampal committed 3 years ago

9bff3704

Fix cudagraph backend (#1393)

xuzhao9 committed 3 years ago

a9771829

Migrate fx2trt and torch_trt to use the backend option. (#1388)

xuzhao9 committed 3 years ago

d5801978

fix bug for short model executions (#1391)

FindHao committed 3 years ago

d166639f

Add support to AITemplate engine (#1386)

xuzhao9 committed 3 years ago

71440ac8

Run 15 rounds and collect the max delta (#1352)

xuzhao9 committed 3 years ago

597a6443

fix bug for record collections (#1366)

FindHao committed 3 years ago

7ab73738

Update the backend interface (#1384)

xuzhao9 committed 3 years ago

eb49327b

remove outdated argument (#1373)

Yueming Hao committed 3 years ago

9b9bcbcb

call eval() for inference tests (#1380)

snadampal committed 3 years ago

410ab490

Remove the use of circleci. (#1258)

xuzhao9 committed 3 years ago

0b8aa0a8

drq: change view to reshape to support channels last (#1367)

chunyuan-w committed 3 years ago

ef39f8ca

Add "cluster" option for slurm cluster settings (#1362)

davidberard98 committed 3 years ago

bbdc7778

Add precision option (#1361)

davidberard98 committed 3 years ago

c65f6cd4

Give more accurate error message (#1358)

brad-mengchi committed 3 years ago

354378b4

Add accuracy metrics to e2e models (#1356)

xuzhao9 committed 3 years ago

1e96f5d3

README: Add `torchaudio` to pip install command (#1355)

mostafaelhoushi committed 3 years ago

71db0c0e

Check the stableness of model tests in userbenchmarks (#1339)

xuzhao9 committed 3 years ago

43ca0857

Fix stray `pretrained=True` warnings from torchvision (#1346)

davidberard98 committed 3 years ago

e1d35191

Install sympy (#1347)

xuzhao9 committed 3 years ago

ff361c69

Add E2E vision_resnet50 model with CIFAR10 input data (#1345)

xuzhao9 committed 3 years ago

b662a6fb

Set cudnn.benchmark in timm and vision models to True by default (#1343)

FindHao committed 3 years ago

fcfc2ba2

Adding bc utility used by memory monitoring (#1342)

weiwangmeta committed 3 years ago

fe46366e

Add switch-cuda utility to docker image (#1341)

xuzhao9 committed 3 years ago

a9dde2ed

Add model phlippe_densenet (#1325)

FindHao committed 3 years ago

c86f9081

Fix the incorrect naming of the scikit-learn deps (#1340)

xuzhao9 committed 3 years ago

34f657e6

Add userbenchmark doc (#1328)

xuzhao9 committed 3 years ago

fab73d18

Add phlippe_resnet (#1324)

FindHao committed 3 years ago

f5f82975

Add doctr recognition and detection models (#1315)

xuzhao9 committed 3 years ago

b99caf81

Newer Older