skywolf829 / GSTK

Gaussian Splatting toolkit application. One stop shop for preprocessing your dataset, training your model with human-in-the-loop training, and editing saved GSplat PLY files.

MIT License

4 stars 1 forks source link

Backend begin/pause/play training #7

Closed skywolf829 closed 7 months ago

skywolf829 commented 7 months ago

A button will be clicked on frontend to begin training, or pause/play if training already started (#4).

Requirements

Check that everything that should be initialized is already (dataset, model, trainer)
Begin training (threading required?)

Extra

Reset training? Reinitializes optimizer, go to iteration 0, but keeps model the same?

skywolf829 commented 7 months ago

Handled with a thread. When training starts, a thread is started that will continue to do train steps. When a stop signal is received, the thread will cut its while loop and join, ending training at that iteration. Training can be resumed simply by starting again, which will start another thread on the train function.