commit 23387f420a49c8bf68e4bb2361bbe84f9d0a2251
author wenqiao zhang <wenqiaozhang@zju.edu.cn> Fri Feb 25 13:21:15 2022 +0800
committer wenqiao zhang <wenqiaozhang@zju.edu.cn> Fri Feb 25 13:21:15 2022 +0800
tree e31afdb2b061f8579228b8fe544c4f62a87a5f4b
parent d6c7375ba203f8ac8005424f915b8f8341af9cbc

update optimizer for bloodmnist

examples/demos/Classification/BloodMnist/ClassDemo.py

diff --git a/examples/demos/Classification/BloodMnist/ClassDemo.py b/examples/demos/Classification/BloodMnist/ClassDemo.py
index 71a0fa6..1d17150 100644
--- a/examples/demos/Classification/BloodMnist/ClassDemo.py
+++ b/examples/demos/Classification/BloodMnist/ClassDemo.py
@@ -246,7 +246,6 @@
 # optimizer_ft = opt.SGD(lr=0.005, momentum=0.9, weight_decay=1e-5, dtype=singa_dtype["float32"])
 optimizer_ft = opt.Adam(lr=1e-3)
 # optim.Adam(filter(lambda p: p.requires_grad, model.parameters()), lr=lr)
-# lr_scheduler = torch.optim.lr_scheduler.ReduceLROnPlateau(optimizer_ft, mode='max', patience=5, threshold=1e-3)
 
 # %% start training
 dev = device.create_cpu_device()