LoRa Training - Local Steps Calculator

详情

模型描述

这是一段你可以本地在电脑上运行的简单 Python 代码。

我一直在为找到 LoRa 训练的正确步数而烦恼,于是决定编写这段简单脚本。我原本是为自己写的,但觉得或许别人也会需要,所以决定分享出来。

你只需要下载这个文件,将其保存在一个安全的文件夹中。每次使用时,只需将其复制粘贴到包含所有训练图像的文件夹中即可。

该脚本会统计图像数量,并给出以下结果:

  • 重复次数

  • 训练轮数(Epochs)

  • 总步数

以下是一些用于保持计算最优的参数:

  • 批次大小固定为 2:这意味着训练时批次始终设为 2。

  • 计算的步数范围固定在 3000 至 5000 之间:因为我了解到 LoRa 训练的“最佳区间”约为 3000 步,所以我让这段脚本自动计算,使步数落在这一范围内。

  • 文件必须与图像放在同一文件夹中:文件夹中可以有其他文件,但脚本只会识别图像文件。

希望这段脚本能帮助你更好地训练 LoRa 模型。

如果你愿意支持我,欢迎打赏,这样我就能在本地训练更多 LoRa 模型了。

谢谢!

此模型生成的图像

未找到图像。