Update README.md
Browse files
README.md
CHANGED
@@ -25,7 +25,7 @@ Please see the GitHub repo for more info: https://github.com/ikergarcia1996/AI-G
|
|
25 |
- No Action conditioning
|
26 |
|
27 |
### dit_action.safetensors
|
28 |
-
- Continue training of `dit.safetensors` with action_conditioning for 210,000 steps.
|
29 |
- Trained using 4xNvidia A100 80Gb in Bfloat16
|
30 |
- 64 batch size
|
31 |
- 1e-4 learning rate with cosine scheduler to 1e-5 and 5% warmup
|
|
|
25 |
- No Action conditioning
|
26 |
|
27 |
### dit_action.safetensors
|
28 |
+
- Continue training of `dit.safetensors` with `action_conditioning` for 210,000 steps.
|
29 |
- Trained using 4xNvidia A100 80Gb in Bfloat16
|
30 |
- 64 batch size
|
31 |
- 1e-4 learning rate with cosine scheduler to 1e-5 and 5% warmup
|