Unverified Commit 52b52c2c authored by guoshzhao's avatar guoshzhao Committed by GitHub
Browse files

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: default avatarGuoshuai Zhao <guzhao@microsoft.com>
parent 9388f8f5
......@@ -11,6 +11,13 @@
from superbench.benchmarks.context import Enum
class Optimizer(Enum):
"""The Enum class representing different optimizers."""
SGD = 'sgd'
ADAM = 'adam'
ADAMW = 'adamw'
class DistributedImpl(Enum):
"""The Enum class representing different distributed implementations."""
DDP = 'ddp'
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment