pytorch/benchmark

Pull Requests Commits

Add explicit APIs in BenchmarkModel

janeyx99 committed 2 years ago

5333e9db

Fix test_opacus_cifar10_train_cpu by removing commas lmao

janeyx99 committed 2 years ago

3181091e

Fix detectron test by moving important error checking

janeyx99 committed 2 years ago

64538fd6

Make sure dependents of optim are also updated

janeyx99 committed 2 years ago

02e9e7f3

Lazily initialize optimizer

janeyx99 committed 2 years ago

2f42ad42

Add get_optimizer and set_optimizer to BenchmarkModel API

janeyx99 committed 2 years ago

bc26c5cd

Add instructions to test model after submitting (#1447)

msaroufim committed 2 years ago

e080877d

GeForce clock query fix Issue #1440 (#1445)

Mike Schneider committed 2 years ago

9ab7a6f2

enable peak memory measurement by default (#1392)

FindHao committed 2 years ago

1604885b

Add the MPS device to eval mode. (#1435)

kulinseth committed 2 years ago

e9941f39

Clarify instllation step (#1426)

spzala committed 2 years ago

f33ce935

Fix the case when workflow is empty (#1438)

xuzhao9 committed 2 years ago

c9e3dd95

Add the --mps_only option. (#1432)

kulinseth committed 2 years ago

1635a749

Disable the nvfuser nightly CI (#1431)

xuzhao9 committed 2 years ago

4fb6838c

Fix torchrec_dlrm model and enable fbcode test (#1428)

xuzhao9 committed 2 years ago

08fc5914

Use the correct batch size for timm ViT to improve stableness (#1419)

xuzhao9 committed 2 years ago

06538fd1

Upgrade to gcc-11 in the A100 Docker to fix CI error (#1427)

xuzhao9 committed 2 years ago

e2dfe592

Add torchtext to fix benchmark installation error (#1424)

spzala committed 2 years ago

0e6a7de9

Upgrade the actions used by the bisection workflow (#1417)

xuzhao9 committed 2 years ago

d2c692c2

replace torchbench impelmentation with torchrec that shards the module (#1401)

colin2328 committed 2 years ago

9806a7ec

Fix vision model on inductor and profile on CPU only machines (#1415)

xuzhao9 committed 2 years ago

7bd50b01

change fastNLP_Bert's task name (#1410)

FindHao committed 2 years ago

3fe6446f

Add detailed task names (#1411)

Yueming Hao committed 2 years ago

74895429

Add channels_last for all torchbench models and enable it with run.py (#1371)

chuanqi129 committed 2 years ago

d64579cc

Pin pygame version to fix soft_actor_critic (#1412)

xuzhao9 committed 2 years ago

43ee38aa

Match AITModule on ait inputs and fx inputs (#262)

tissue3 committed 2 years ago

1bbb66c4

Fix type in iter rounds (#1408)

xuzhao9 committed 2 years ago

411117d8

Add mimo_cmf_30x

bertmaher committed 2 years ago

bf02c40a

Upgrade to upload-artifact v3 to resolve GitHub warnings (#1407)

xuzhao9 committed 2 years ago

91d91104

Force creation of the conda environment and upgrade dependencies (#1406)

xuzhao9 committed 2 years ago

adce3da2

Older