[lab4] Minor fixes

lab4/src/main.rs

@@ -89,7 +89,8 @@ impl eframe::App for MyApp {
                             epochs_left: self.epochs_count,
                             epochs_per_update: min(10, self.epochs_count / 60),
                             n: 0.0,
-                        }
+                        };
+                        self.model.set_random_weights(&mut rand::rng());
                     }
 
                     match &mut self.training {
@@ -180,5 +181,9 @@ impl eframe::App for MyApp {
                     });
                 });
         });
+
+        if matches!(self.training, TrainingState::Training {..}) {
+            ui.request_repaint()
+        }
     }
 }