diff --git a/examples/benchmarks/bert/main.py b/examples/benchmarks/bert/main.py index 277eee2d4..15ccf3868 100644 --- a/examples/benchmarks/bert/main.py +++ b/examples/benchmarks/bert/main.py @@ -122,7 +122,7 @@ def build_optimizer(cfg, model): if cfg.name == 'decoupled_adamw': return DecoupledAdamW(model.parameters(), lr=cfg.lr, - betas=cfg.betas, + betas=list(cfg.betas), eps=cfg.eps, weight_decay=cfg.weight_decay) else: