From 8b6f36a8c95653eef85d19529a657d6c2133b5a3 Mon Sep 17 00:00:00 2001
From: deepr <hexiangdong2020@outlook.com>
Date: Tue, 20 Jul 2021 22:58:06 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BD=BF=E7=94=A8=E5=8F=AF=E5=8F=98=E5=AD=A6?=
 =?UTF-8?q?=E4=B9=A0=E7=8E=87?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 main.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/main.py b/main.py
index 10cd0e1..f4ad80b 100644
--- a/main.py
+++ b/main.py
@@ -28,7 +28,7 @@ train_dataset = train_dataset.batch(4, drop_remainder=True)
 lr_iter = exponential_lr(3e-5, 20, 0.98, 500, staircase=True)
 
 net_loss = SoftmaxCrossEntropyLoss(6, 255)
-net_opt = nn.Adam(net.trainable_params(), learning_rate=3e-5)
+net_opt = nn.Adam(net.trainable_params(), learning_rate=lr_iter)
 
 config_ck = CheckpointConfig(save_checkpoint_steps=1875, keep_checkpoint_max=10)
 ckpoint = ModelCheckpoint(prefix="checkpoint_lenet", config=config_ck)
-- 
GitLab