作者: QIU Tian
机构: 浙江大学
English | 简体中文
- 创建
your_optimizer.py。 - 在
your_optimizer.py中，定义你的优化器。
# your_optimizer.py
__all__ = ['YourOptimizer']
...
class YourOptimizer(Optimizer):
...-
在
__init__.py中，-
导入你的优化器。
-
在
build_optimizer()中注册你的优化器。
-
# __init__.py
...
from .your_optimizer import YourOptimizer
def build_optimizer(args, params):
optimizer_name = args.optimizer.lower()
...
if optimizer_name == 'your_optimizer':
return YourOptimizer(params, ...)
...- 当使用你的优化器时，把
--optimizer赋值为你的优化器名称your_optimizer。注意your_optimizer不需要和你的优化器类名YourOptimizer保持一致。