LoRA Training Calculator for Colab

I spent quite a while toiling over calculations for how many steps it would take for my LoRAs to cook and, by extension, how long it would be, until I decided to make a sheet for it and then, I realized how useful it would be to share it with others. So, here you all go!

The Parameters are as follows (I'll repost this in the tool page itself for future reference):

These ones you have to set manually:

• Images: How many images you have in your dataset.

• Repeats: How many repeats you have.

• Epochs: How many Epochs you have.

• Batch Size: The Batch Size you have set in the Lora trainer.

• Override: Only set this up if the expected auto-calculated steps don't line up with the steps you're seeing in the Colab (ex. if you have 1250 Steps but the Colab says it's training 1300 Steps, you put 1300 in the Override)

• IT/s: What is the overall rate of which your LoRA is currently being trained at.

• Epochs Remaining: How many Epochs are left until your LoRA is complete.

From there, it will auto calculate the rest of the steps so you don't need to worry about setting those up.

• Steps: It will give you an estimated number of steps that the LoRA will take to complete.

• Steps Remaining: It will give you an estimated number of steps remaining for the LoRA to be completed.

• Time Remaining: It will give you an estimated time before the LoRA is completed in seconds, minutes, and hours.

• Difference: The difference between the auto-calculated Steps and the Overridden Steps. The bottom row will show the entire differences.