<img width="1034" height="316" alt="Image" src="https://github.com/user-attachments/assets/aecdce3d-c2b6-4a53-8580-66d17fc1b3dc" /> The initial model seems to take longer to load as the step number in which the run starts get larger. Perhaps we are iterating over the entire dataset to set the correct step number. The correlation seems to be linear.