| .. |
|
__init__.py
|
aa24e8f57e
fix typos
|
hai 1 ano |
|
config_utils.py
|
c18a0d277f
changed dataset to ocrvqa
|
hai 11 meses |
|
dataset_utils.py
|
ce299b3439
add get_custom_data_collator feature
|
hai 1 ano |
|
flop_utils.py
|
26e877fd42
changed readme, unified the context interface and added get_flops_per_sec()
|
hai 1 ano |
|
fsdp_utils.py
|
1a76080807
lora+fsdp working
|
hai 1 ano |
|
memory_utils.py
|
c5a382e509
Make tests run on cpu only machines
|
hai 1 ano |
|
plot_metrics.py
|
6d449a859b
New folder structure (#1)
|
hai 1 ano |
|
train_utils.py
|
d31ee18e4f
Print model's frozen status after freezing
|
hai 10 meses |
JimChienTW
d31ee18e4f
Print model's frozen status after freezing