Skip to content

Commit 1b12c4a

Browse files
committed
Set better default LR across un/supervised
1 parent 53dabb5 commit 1b12c4a

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

napari_cellseg3d/code_plugins/plugin_model_training.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -443,11 +443,15 @@ def _toggle_unsupervised_mode(self, enabled=False):
443443
self.start_btn = self.start_button_unsupervised
444444
self.image_filewidget.text_field.setText("Validation images")
445445
self.labels_filewidget.text_field.setText("Validation labels")
446+
self.learning_rate_choice.lr_value_choice.setValue(1)
447+
self.learning_rate_choice.lr_exponent_choice.setCurrentIndex(1)
446448
else:
447449
unsupervised = False
448450
self.start_btn = self.start_button_supervised
449451
self.image_filewidget.text_field.setText("Images directory")
450452
self.labels_filewidget.text_field.setText("Labels directory")
453+
self.learning_rate_choice.lr_value_choice.setValue(2)
454+
self.learning_rate_choice.lr_exponent_choice.setCurrentIndex(3)
451455

452456
supervised = not unsupervised
453457
self.unsupervised_mode = unsupervised

0 commit comments

Comments
 (0)