waveshaker commited on
Commit
fd889ac
Β·
verified Β·
1 Parent(s): 93f1d40

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +16 -4
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 (`Gradient Accumulation: 3`) β†’ ~0.83s/img (36m 17s)
 
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+)