From 88f5518f92a3e459bb70cc0aa54ec31e8db96355 Mon Sep 17 00:00:00 2001
From: Aaron Gokaslan <aaronGokaslan@gmail.com>
Date: Fri, 1 Mar 2024 08:15:50 -0800
Subject: [PATCH] Fix betas bug

---
 examples/benchmarks/bert/main.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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: