LoRA Training Calculator for Colab

LoRAのトレーニングに必要なステップ数を計算するのに、かなりの時間を費やしました。そして、その結果として、それをひとつのシートにまとめたところ、他の人にも共有すれば役立つことに気づきました。それでは、皆さんへ！

パラメータは以下の通りです（今後のために、この内容をツールページにも再掲します）：

以下は手動で設定する必要があります：

• 画像数：データセットに含まれる画像の枚数。
• リピート回数：繰り返し回数。
• エポック数：設定したエポック数。
• バッチサイズ：LoRAトレーナーで設定したバッチサイズ。
• 上書き：自動計算されたステップ数がColabで表示されているステップ数と一致しない場合にのみ設定してください（例：予想ステップが1250だが、Colabが1300ステップと表示している場合、上書き欄に1300を入力）。
• IT/s：現在のLoRAのトレーニング速度（1秒あたりの処理ステップ数）。
• 残りエポック数：LoRAのトレーニングが完了するまでに残っているエポック数。

これにより、残りのステップ数は自動計算されるため、手動で設定する必要はありません。

• ステップ数：LoRAのトレーニングに必要な総ステップ数の推定値を表示します。
• 残りステップ数：LoRAのトレーニング完了までに残っているステップ数の推定値を表示します。
• 残り時間：LoRAのトレーニングが完了するまでの推定時間（秒、分、時間単位）を表示します。
• 差分：自動計算されたステップ数と上書きされたステップ数との差。最下段には全体の差分が表示されます。