Safetensors
English
Zhiminli commited on
Commit
e13c61b
1 Parent(s): 1c074f9

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +12 -0
README.md CHANGED
@@ -85,6 +85,18 @@ PYTHONPATH=./ deepspeed hydit/train_large_deepspeed.py \
85
  "$@"
86
  ```
87
 
 
 
 
 
 
 
 
 
 
 
 
 
88
  ## Inference
89
 
90
  ### Using Gradio
 
85
  "$@"
86
  ```
87
 
88
+ 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
+
100
  ## Inference
101
 
102
  ### Using Gradio