| .. |
|
__init__.py
|
aa24e8f57e
fix typos
|
2 роки тому |
|
config_utils.py
|
c18a0d277f
changed dataset to ocrvqa
|
1 рік тому |
|
dataset_utils.py
|
ce299b3439
add get_custom_data_collator feature
|
1 рік тому |
|
flop_utils.py
|
26e877fd42
changed readme, unified the context interface and added get_flops_per_sec()
|
1 рік тому |
|
fsdp_utils.py
|
1a76080807
lora+fsdp working
|
1 рік тому |
|
memory_utils.py
|
c5a382e509
Make tests run on cpu only machines
|
1 рік тому |
|
plot_metrics.py
|
6d449a859b
New folder structure (#1)
|
1 рік тому |
|
train_utils.py
|
5fdeb55058
chore: update train_utils.py (#690)
|
1 рік тому |
Sanyam Bhutani
5fdeb55058
chore: update train_utils.py (#690)