| .. |
|
configs
|
cc356b6017
Add FSDP CPU offloading option
|
1 year ago |
|
data
|
63ce4ce7f6
Moved sampler to data submodule
|
1 year ago |
|
datasets
|
63ce4ce7f6
Moved sampler to data submodule
|
1 year ago |
|
inference
|
e554c1c8bf
The tokenizer will not add eos_token by default
|
1 year ago |
|
model_checkpointing
|
ce9501f22c
remove relative imports
|
1 year ago |
|
policies
|
ce9501f22c
remove relative imports
|
1 year ago |
|
utils
|
40b32ba559
Fix tqdm bar not change length after terminal is resized (#201)
|
1 year ago |
|
finetuning.py
|
91e2573aa8
pass weight_decay into optimizer
|
1 year ago |
Matthias Reso
63ce4ce7f6
Moved sampler to data submodule