Fix for black

This commit is contained in:
Yifan Yang 2023-06-02 17:07:56 +08:00
parent 274a1ef45f
commit 4f152327c6

View File

@ -667,7 +667,7 @@ def run(rank, world_size, args):
model.parameters(), model.parameters(),
lr=params.lr, lr=params.lr,
weight_decay=params.weight_decay, weight_decay=params.weight_decay,
) )
scheduler = NewBobScheduler( scheduler = NewBobScheduler(
optimizer, optimizer,