config: Use Adam optimizer to UNet

synml · synml · commit e881a2a7c313 · 2021-02-05T23:13:52.000+09:00
diff --git a/config.yaml b/config.yaml
@@ -11,9 +11,8 @@ UNet:
   batch_size: 16
   epoch: 100
   optimizer:
-    name: SGD
+    name: Adam
     lr: 0.001
-    momentum: 0.9
     weight_decay: 0.00001
   scheduler:
     name: ReduceLROnPlateau
