From 9c75584e268a47ef15501e90d88107135e08c16c Mon Sep 17 00:00:00 2001
From: Junhong Xu <jxu7@bsu.edu>
Date: Thu, 8 Jun 2017 23:06:32 -0400
Subject: [PATCH] DenseNet161 with SGD

---
 trainers/train_densenet.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/trainers/train_densenet.py b/trainers/train_densenet.py
index f53bd99..f184b82 100644
--- a/trainers/train_densenet.py
+++ b/trainers/train_densenet.py
@@ -46,7 +46,7 @@ def train(epoch):
     criterion = MultiLabelSoftMarginLoss()
     net = densenet169(pretrained=False)
     logger = Logger('../log/', NAME)
-    optimizer = optim.SGD(lr=1e-1, params=net.parameters(), weight_decay=5e-4, momentum=0.8, nesterov=True)
+    optimizer = optim.SGD(lr=1e-1, params=net.parameters(), weight_decay=5e-4, momentum=0.9, nesterov=True)
     # optimizer = get_optimizer(net, False, 1e-4, 1e-4)
     # optimizer = optim.Adam(params=net.parameters(), lr=5e-4, weight_decay=5e-4)
     net.cuda()