| .. |
|
configs
|
cc356b6017
Add FSDP CPU offloading option
|
2 years ago |
|
datasets
|
6217635e87
Fix tqdm bar not change length after terminal is resized
|
2 years ago |
|
inference
|
e554c1c8bf
The tokenizer will not add eos_token by default
|
2 years ago |
|
model_checkpointing
|
ce9501f22c
remove relative imports
|
2 years ago |
|
policies
|
ce9501f22c
remove relative imports
|
2 years ago |
|
utils
|
92be45b0fe
Adding matplotlib to requirements. Removing import from train_utils
|
2 years ago |
|
finetuning.py
|
91e2573aa8
pass weight_decay into optimizer
|
2 years ago |
Beto
92be45b0fe
Adding matplotlib to requirements. Removing import from train_utils