OneTrainer icon indicating copy to clipboard operation
OneTrainer copied to clipboard

Validation implementing

Open seniorsolt opened this issue 7 months ago • 2 comments

Simple validation logic was implemented

ValidationBaseDataLoader class is added. This class is primarily a clone of StableDiffusionBaseDataLoader with an added boolean variable validation in the create_dataset method.

To process new variable and divide train and validation datasets DataLoaderMgdsMixin was modified. UI was updated to provide radio buttons for validation logic.

It's hard to consider that PR ready to merge. I suggest we discuss this further to ensure alignment with your future plans.

seniorsolt avatar Jul 03 '24 17:07 seniorsolt