| .. |
|
__init__.py
|
aa24e8f57e
fix typos
|
1 anno fa |
|
config_utils.py
|
c18a0d277f
changed dataset to ocrvqa
|
11 mesi fa |
|
dataset_utils.py
|
ce299b3439
add get_custom_data_collator feature
|
11 mesi fa |
|
flop_utils.py
|
26e877fd42
changed readme, unified the context interface and added get_flops_per_sec()
|
1 anno fa |
|
fsdp_utils.py
|
1a76080807
lora+fsdp working
|
11 mesi fa |
|
memory_utils.py
|
c5a382e509
Make tests run on cpu only machines
|
1 anno fa |
|
plot_metrics.py
|
6d449a859b
New folder structure (#1)
|
1 anno fa |
|
train_utils.py
|
d6ae2031c3
Fix fine-tuning training loss accumulation (#725)
|
10 mesi fa |
celestinoalan
d6ae2031c3
Fix fine-tuning training loss accumulation (#725)