- 22 Mar, 2021 2 commits
-
-
guoshzhao authored
* move benchmarks registration from registry.py to __init__.py * revise __init__. Co-authored-by:Guoshuai Zhao <guzhao@microsoft.com>
-
guoshzhao authored
Benchmarks: Add Feature - Add benchmark finish check according to num_warmup/num_steps and duration in ModelBenchmark class. (#25) * add is_finished function * reuse current time. Co-authored-by:
-
- 18 Mar, 2021 3 commits
-
-
guoshzhao authored
Benchmarks: Add Feature - Register all benchmarks by import model_benchmarks/micro_benchmarks/docker_benchmarks. (#21) * clean code. * minor change. Co-authored-by: -
guoshzhao authored
* add sample_count argument. * handle more condidatins. * address comments. Co-authored-by: -
guoshzhao authored
* support benchmark re-registration. * address comments Co-authored-by:
-
- 15 Mar, 2021 1 commit
-
-
guoshzhao authored
Co-authored-by:
-
- 11 Mar, 2021 1 commit
-
-
guoshzhao authored
* add random dataset. * install pytorch-cpu for test docker. * fix typo * add more test cases. * address comments. Co-authored-by:
-
- 09 Mar, 2021 2 commits
-
-
guoshzhao authored
* add flag to disable GPU. * fix spelling * fix unittest. * address comments. Co-authored-by: -
guoshzhao authored
Co-authored-by:
-
- 08 Mar, 2021 2 commits
-
-
guoshzhao authored
* add pytorch base class * address comments Co-authored-by: -
guoshzhao authored
* add optimizer definition and function to create torch optimizer. * move optimizer enum into model_base module. Co-authored-by:
-
- 04 Mar, 2021 2 commits
-
-
guoshzhao authored
Co-authored-by: -
guoshzhao authored
Co-authored-by:
-
- 24 Feb, 2021 1 commit
-
-
guoshzhao authored
* benchmarks init. Co-authored-by:
-
