Update README.md
Browse files
README.md
CHANGED
@@ -89,11 +89,11 @@ Recommended parameter settings
|
|
89 |
|
90 |
| Parameter | Description | Recommended Parameter Value | Note|
|
91 |
|:---------------:|:---------:|:---------------------------------------------------:|:--:|
|
92 |
-
| `--batch_size` | Training batch size | 1 | Depends on GPU memory|
|
93 |
-
| `--grad-accu-steps` | Size of gradient accumulation | 2 | |
|
94 |
| `--rank` | Rank of lora | 64 | 8-128 are all possible|
|
95 |
| `--max-training-steps` | Training steps | 2000 | Varies with the amount of training data, about 2000 steps are enough for 100 images|
|
96 |
-
| `--lr` | Learning rate | 0.0001 |
|
97 |
|
98 |
|
99 |
|
|
|
89 |
|
90 |
| Parameter | Description | Recommended Parameter Value | Note|
|
91 |
|:---------------:|:---------:|:---------------------------------------------------:|:--:|
|
92 |
+
| `--batch_size` | Training batch size | 1 | Depends on GPU memory|
|
93 |
+
| `--grad-accu-steps` | Size of gradient accumulation | 2 | - |
|
94 |
| `--rank` | Rank of lora | 64 | 8-128 are all possible|
|
95 |
| `--max-training-steps` | Training steps | 2000 | Varies with the amount of training data, about 2000 steps are enough for 100 images|
|
96 |
+
| `--lr` | Learning rate | 0.0001 | - |
|
97 |
|
98 |
|
99 |
|