Lengyue
|
2733ed600f
Accelerate training with fsdp
|
2 年之前 |
Lengyue
|
01cf9bbafd
Fix dataset and training loop
|
2 年之前 |
Lengyue
|
0282451dff
fix scheduler
|
2 年之前 |
Lengyue
|
7650b2ac43
Update much robust dataloader
|
2 年之前 |
Lengyue
|
60ac11d7fe
Update deps
|
2 年之前 |
Lengyue
|
0dae48d2dc
Fix model training mode
|
2 年之前 |
Lengyue
|
3297b2dc2c
Add hubert vq training
|
2 年之前 |
Lengyue
|
900fc9dec9
Optimize data loading & gradient accumulate
|
2 年之前 |
Lengyue
|
026d419e9a
Fix logger in config
|
2 年之前 |
Lengyue
|
db347ca973
Add llm exp code
|
2 年之前 |
Lengyue
|
9adf687fb1
Update base training code
|
2 年之前 |
Lengyue
|
aa0a6be778
init training code
|
2 年之前 |