pytorch/benchmark

Pull Requests Commits

Changed the algorithm to GlobalNoUTurnSampler

agunapal committed 3 years ago

41b18a7f

Added a comment regarding num_warmup

agunapal committed 3 years ago

924c75c8

Update gen_inputs to support both train & eval

agunapal committed 3 years ago

ef3c5dcf

Updated gen_inputs to generate new data for each batch

agunapal committed 3 years ago

f66856b9

Fixed gen_inputs interface

agunapal committed 3 years ago

3dff318c

Disabling jit test as torch script cannot handle this model

agunapal committed 3 years ago

8746c63c

added interface for gen_inputs

agunapal committed 3 years ago

3dd27d50

Reduced the eval computation to prevent timeout on CI runner

agunapal committed 3 years ago

8d203f0d

increased evaluation compute time

agunapal committed 3 years ago

5bc622c0

Updated comments for train method

agunapal committed 3 years ago

5ba8a66a

Added train functionality

agunapal committed 3 years ago

30e3a7c5

Increased number of iterations based on sample example, eval batch size is not adjustable

agunapal committed 3 years ago

17116f0d

Use torch.nn.Module wrapper

agunapal committed 3 years ago

77fea82d

Return Tensor object

agunapal committed 3 years ago

991fda83

added yaml file. Need to confirm if the values are set correctly

agunapal committed 3 years ago

0dd160d8

Added init and eval method for RobustRegression

agunapal committed 3 years ago

35c4dfba

added install code

agunapal committed 3 years ago

2022eb61

Fix PyTorch unet train, only use mixed precision in forward pass. (#873)

xuzhao9 committed 3 years ago

6b13e204

Add more NVFuser microbenchmarks (#801)

davidberard98 committed 3 years ago

366b1843

Migrate from Ubuntu 16.04 to 20.04 (#765)

kit1980 committed 3 years ago

07a4914c

move torchrec dlrm example into facebookresearch/dlrm (#221)

colin2328 committed 3 years ago

c5b02e7d

Delete dead Py2 code from pytorch/benchmark (#866)

lisroach committed 3 years ago

acac953b

Add CI test on compiler backends (#832)

xuzhao9 committed 3 years ago

e990e696

Fix demucs jit error with type annotation (#856)

xuzhao9 committed 3 years ago

fbd17497

adding lazy_bench tests (#846)

Krovatkin committed 3 years ago

7c6da287

nvfuser microbenches - import utils from torch.utils (#854)

davidberard98 committed 3 years ago

c4710504

Porting fixes from wconstab/ltc to main (#851)

Krovatkin committed 3 years ago

c60f7edd

Add FAMBench xlmr model (#835)

xuzhao9 committed 3 years ago

032f079f

Improve the `run_sweep.py` script to support precision attribute (#844)

xuzhao9 committed 3 years ago

356fd5da

Add accelerate to requirements.txt (#849)

mrshenli committed 3 years ago

86d29d08

Older