changed add layer combobox to submenu

uek-1 · Jul 7, 2023 · 7cb6821 · 7cb6821
1 parent d781bf6
commit 7cb6821
Showing 1 changed file with 9 additions and 8 deletions.
diff --git a/model-gui/src/widgets/model.rs b/model-gui/src/widgets/model.rs
@@ -73,14 +73,15 @@ fn layer_context_menu(ui : &mut egui::Ui, layer_state : &mut LayerState, layer_m
             .hint_text("Layer Size")
             .show(ui);
 
-        egui::ComboBox::from_label("Activation Function")
-            .selected_text(format!("{:?}", layer_state.activation))
-            .show_ui(ui, |ui| {
-                ui.selectable_value(&mut layer_state.activation, Activation::None, "None");
-                ui.selectable_value(&mut layer_state.activation, Activation::Sigmoid, "Sigmoid");
-                ui.selectable_value(&mut layer_state.activation, Activation::Relu, "ReLu");
-                ui.selectable_value(&mut layer_state.activation, Activation::Softmax, "Softmax");
-            });
+        //egui::ComboBox::from_label("Activation Function")
+            //.selected_text(format!("{:?}", layer_state.activation))
+        ui.menu_button("Activation", |ui| { 
+            ui.selectable_value(&mut layer_state.activation, Activation::None, "None");
+            ui.selectable_value(&mut layer_state.activation, Activation::Sigmoid, "Sigmoid");
+            ui.selectable_value(&mut layer_state.activation, Activation::Relu, "ReLu");
+            ui.selectable_value(&mut layer_state.activation, Activation::Softmax, "Softmax");
+        });
+
 
         if ui.button("Add").clicked() {
             let input_shape = layer_menu_state.input_shape;