commit ee2e46050deb5060664881f35e8584326936dc4c
author Zhaojing Luo <zhaojing@comp.nus.edu.sg> Wed Mar 02 16:23:16 2022 +0800
committer GitHub <noreply@github.com> Wed Mar 02 16:23:16 2022 +0800
tree c46f3f336a0352d4c2dce08a49892b2fd3a21dfb
parent cd1d57564b805e95b115e60b389e35191e2a34a5
parent 6d840e48f385b1fcc1601dfe43fe8a7d3ea5736e

Merge pull request #937 from zlheui/update-bloodmnist-optimizer

replace SGD optimizer with Adam 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 1d17150..1047c21 100644
--- a/examples/demos/Classification/BloodMnist/ClassDemo.py
+++ b/examples/demos/Classification/BloodMnist/ClassDemo.py
@@ -243,7 +243,6 @@
 # %% model configuration
 model = CNNModel(num_classes=num_class)
 criterion = layer.SoftMaxCrossEntropy()
-# 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)