| .. |
|
__init__.py
|
aa24e8f57e
fix typos
|
hace 1 año |
|
config_utils.py
|
c18a0d277f
changed dataset to ocrvqa
|
hace 11 meses |
|
dataset_utils.py
|
ce299b3439
add get_custom_data_collator feature
|
hace 1 año |
|
flop_utils.py
|
26e877fd42
changed readme, unified the context interface and added get_flops_per_sec()
|
hace 1 año |
|
fsdp_utils.py
|
1a76080807
lora+fsdp working
|
hace 11 meses |
|
memory_utils.py
|
c5a382e509
Make tests run on cpu only machines
|
hace 1 año |
|
plot_metrics.py
|
6d449a859b
New folder structure (#1)
|
hace 1 año |
|
train_utils.py
|
2a94bfff26
Append epoch rather than best val. loss to val_loss
|
hace 10 meses |
celestinoalan
2a94bfff26
Append epoch rather than best val. loss to val_loss