From 2dcf1254cf8c112dfc1c1a17f385e2ae4b3240b3 Mon Sep 17 00:00:00 2001 From: XingyuXie Date: Fri, 2 Sep 2022 13:44:13 +0800 Subject: [PATCH] fix typo --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index 1b6a463..3e117a0 100644 --- a/README.md +++ b/README.md @@ -43,7 +43,7 @@ In all experiments, we set `no-prox=False` in our paper. **Step 2.** creat the Adan optimizer as follows. In this step, we can directly replace the vanilla optimizer by using the following command: ```python -from adam import Adan +from adan import Adan optimizer = Adan(param, lr=args.lr, weight_decay=args.weight_decay, betas=args.opt_betas, eps = args.opt_eps, max_grad_norm=args.max_grad_norm, no_prox=args.no_prox) ```