sd-scripts icon indicating copy to clipboard operation
sd-scripts copied to clipboard

wandb logging parameters error: Dataset_Repeats not correct

Open rafstahelin opened this issue 1 year ago • 3 comments

I have my training images set to 10 repeats using network training script to 10_token class, but when logging to WandB the run shows dataset_repeat = 1. Might be logging the regularization repeats. Still this is wrong

rafstahelin avatar Jul 01 '24 09:07 rafstahelin

This is probably because it's capturing the argument value for the script args.dataset_repeat but your value is part of the datasets subset which may be in a different place. Since this supports multiple datasets/subsets it might be tricky to get this extracted effectively to a single value.

It may be appropriate to extract it if it's a single dataset value though.

rockerBOO avatar Jul 01 '24 20:07 rockerBOO

@rockerBOO how do I extract it? I usually only use one dataset no subsets (until now). Also, would be good to get the path and folder used for the training data. I don't see it in wandb either

rafstahelin avatar Jul 03 '24 08:07 rafstahelin

Under ss_dataset_dirs there is n_repeats in the json.

rockerBOO avatar Jul 03 '24 18:07 rockerBOO