Benchmarks: Add Feature - Add optimizer definition in Model Base (#13)
* add optimizer definition and function to create torch optimizer.
* move optimizer enum into model_base module.
Co-authored-by:
Guoshuai Zhao <guzhao@microsoft.com>
Showing
Please register or sign in to comment