superbench/benchmarks/model_benchmarks/model_base.py · 52b52c2cbcc68c501c440e742741969e1117261d · tsoc / superbenchmark

Benchmarks: Add Feature - Add optimizer definition in Model Base (#13) · 52b52c2c

guoshzhao authored Mar 08, 2021



* add optimizer definition and function to create torch optimizer.

* move optimizer enum into model_base module.
Co-authored-by: Guoshuai Zhao <guzhao@microsoft.com>

52b52c2c

model_base.py 11.3 KB

Replace model_base.py