Update README.md
Browse files
README.md
CHANGED
|
@@ -57,13 +57,25 @@ pipeline_tag: text-to-image
|
|
| 57 |
- Presets: `Moss-Image Fast(rank 16, 0.25MP)`
|
| 58 |
- Pictrue Size: `0.26 MP` β 512 res
|
| 59 |
- Training Speed (Benchmark: 39 imgs, 0.26MP, r16):
|
| 60 |
-
- OneTrainer: 910 steps, 1.12 s/it (`Local Batch Size: 3`) β ~0.37s/img (17m 43s)
|
| 61 |
-
- Fizgig(Batch-Size: Not Supported): 2730 steps, 1.25 it/s (`Gradient Accumulation: 3`) β ~0.83s/img (36m 17s)
|
| 62 |
-
- Fizgig-Advanced(Batch-Size: Supported): 2730 steps, 1.25 it/s (`
|
|
|
|
| 63 |
- Showcase & Card Image Parameters:
|
| 64 |
- Model: -
|
| 65 |
- LoRA Weight: 0.8 ~ 1
|
| 66 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 67 |
- **Krea2**
|
| 68 |
- Training Tools: [Onetrainer](https://github.com/nerogar/OneTrainer)
|
| 69 |
- Dataset: Mostly 50+ images (averaging 100+)
|
|
|
|
| 57 |
- Presets: `Moss-Image Fast(rank 16, 0.25MP)`
|
| 58 |
- Pictrue Size: `0.26 MP` β 512 res
|
| 59 |
- Training Speed (Benchmark: 39 imgs, 0.26MP, r16):
|
| 60 |
+
- OneTrainer: 910 steps, 1.12 s/it (`Local Batch Size: 3, LR: 0.0003`) β ~0.37s/img (17m 43s)
|
| 61 |
+
- Fizgig(Batch-Size: Not Supported): 2730 steps, 1.25 it/s (`Gradient Accumulation: 3, LR: 0.0001`) β ~0.83s/img (36m 17s)
|
| 62 |
+
- Fizgig-Advanced(Batch-Size: Supported): 2730 steps, 1.25 it/s (`Batch Size: 3, LR: 0.0003`) β ~0.83s/img (36m 17s)
|
| 63 |
+
- *Note on Speed & Formats: OneTrainer used the official sharded Diffusers format (bfloat16), while Fizgig used the ComfyUI single-file format (int8).*
|
| 64 |
- Showcase & Card Image Parameters:
|
| 65 |
- Model: -
|
| 66 |
- LoRA Weight: 0.8 ~ 1
|
| 67 |
+
|
| 68 |
+
> **π‘ Quick Note: LR / Batch Size / Gradient Accumulation**
|
| 69 |
+
>
|
| 70 |
+
> - **Effective Batch Size** = `Batch Size Γ Gradient Accumulation`. Gradient accumulation saves VRAM but is slower; a larger Batch Size is faster but uses more VRAM.
|
| 71 |
+
> - **LR Scaling**: Don't blindly apply the linear rule to LoRA. Validate with loss curves. (My baseline: `Batch 3` + `LR 0.0003`). *Note: The vanilla Fizgig benchmark used the default `LR 0.0001`, which is technically under-scaled for an effective batch of 3; however, it was sufficient for this speed test.*
|
| 72 |
+
>
|
| 73 |
+
> **Paper TL;DR:**
|
| 74 |
+
> - [*Accurate, Large Minibatch SGD* (2017)](https://arxiv.org/abs/1706.02677): The "Linear Scaling Rule" β if you multiply batch size by k, multiply LR by k. This was designed for full fine-tuning on ImageNet, not LoRA.
|
| 75 |
+
> - [*One weird trick for parallelizing CNNs* (2014)](https://arxiv.org/abs/1404.5997): The "Square Root Scaling Rule" β to keep gradient variance stable, scale LR by βk instead. More conservative when batch gets very large.
|
| 76 |
+
> - [*Beware of the Batch Size* (2026)](https://arxiv.org/abs/2602.09492): For LoRA specifically, batch size is not just a multiplier for LR β it's a **first-order design parameter**. Optimal batch depends on dataset size (most sensitive), rank, and model capacity. Vanilla LoRA matches fancy variants when batch is tuned properly.
|
| 77 |
+
|
| 78 |
+
|
| 79 |
- **Krea2**
|
| 80 |
- Training Tools: [Onetrainer](https://github.com/nerogar/OneTrainer)
|
| 81 |
- Dataset: Mostly 50+ images (averaging 100+)
|