Use of Adam optimizer for segmentation

Belval · Dec 7, 2019 · 98036f8 · 98036f8
1 parent 5b79a3f
commit 98036f8
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 4 deletions.
diff --git a/src/train.py b/src/train.py
@@ -137,8 +137,7 @@ def train():
         )
         model.compile(
             loss=jaccard_distance,
-            # loss=dice_coef_multilabel_builder(args.class_count),
-            optimizer=RMSprop(lr=args.learning_rate, momentum=0.9),
+            optimizer=Adam(lr=args.learning_rate),
             metrics=["accuracy"],
         )
     elif args.task == 'classification':

diff --git a/src/utils.py b/src/utils.py
@@ -46,5 +46,3 @@ def parse_labelme_file(sample_path):
             mask[fill_row_coords, fill_col_coords] = 255
             masks.append((int(shape["label"]), mask))
     return arr, masks
-
-