| .. |
|
configs
|
0c084ec779
Apply gradient accumulate to save memory
|
2 rokov pred |
|
datasets
|
a003e5a390
Add utilities & whisper vq model
|
2 rokov pred |
|
models
|
086c82d789
Fix vq whisper training & remove ema in tb logger
|
2 rokov pred |
|
utils
|
7650b2ac43
Update much robust dataloader
|
2 rokov pred |
|
init_model.py
|
9adf687fb1
Update base training code
|
2 rokov pred |
|
logger.py
|
3297b2dc2c
Add hubert vq training
|
2 rokov pred |
|
scheduler.py
|
3297b2dc2c
Add hubert vq training
|
2 rokov pred |
|
train.py
|
4cfb62a08a
Fix gradient accumulate logging
|
2 rokov pred |
Lengyue
0c084ec779
Apply gradient accumulate to save memory