diffusion-visualizations icon indicating copy to clipboard operation
diffusion-visualizations copied to clipboard

Feature: Reset Training Button

Open helblazer811 opened this issue 8 months ago • 0 comments

Summary: Add a button to the left of the run training button that resets the model.

Styling:

  • Have a button like this.
  • Set the height of the button to the same height as the epochs counter text.
  • Put it to the left of the run training button.

State Changes:

  • The button should set epochs to zero.
  • When training, resetting should stop training after it sets it to zero.
  • The run training button should resume training, rather than resetting after the button.
  • If a pretrained model is selected, the button should switch to non-pretrained and reset that model to zero-epochs (if there is one).

Implementation Details:

  • Add a reset_training function to the state_mangagement.ts file.
  • Add a ResetTrainingButton.svelte component, that on-click runs the reset_training function

helblazer811 avatar May 31 '25 19:05 helblazer811