[FEATURE ENHANCEMENT] Custom Steps Per Epoch in GUI

[adamcatto]

I have a weakly-supervised learning pipeline in which I generate predictions for all frames in my dataset (~2 million) and filter out frames that do not meet certain QC, such as tracking score and anatomical priors on collections of body parts. I then train another model on the frames that pass these criteria (~135k frames), but currently I need to wait > 8 hours for one epoch to finish. I saw that the dataset class sets a steps_per_epoch attribute; I would love to have a way to customize the steps per epoch in the GUI if possible.

[gitttt-1234]

Hi @adamcatto !

We could add the feature in the GUI in the next release. Until then, you can set the trainer_config. train_steps_per_epoch parameter in the config file if you're using CLI!

Let us know if you have any questions!

Thanks,

Divya