| .. |
|
configs
|
cc356b6017
Add FSDP CPU offloading option
|
1 vuosi sitten |
|
datasets
|
b8d327f7c3
Update grammar_dataset_process.ipynb
|
1 vuosi sitten |
|
inference
|
e554c1c8bf
The tokenizer will not add eos_token by default
|
1 vuosi sitten |
|
model_checkpointing
|
ce9501f22c
remove relative imports
|
1 vuosi sitten |
|
policies
|
ce9501f22c
remove relative imports
|
1 vuosi sitten |
|
utils
|
33925f71e6
Add missing amp context if use_fp16 is enabled
|
1 vuosi sitten |
|
finetuning.py
|
91e2573aa8
pass weight_decay into optimizer
|
1 vuosi sitten |
Matthias Reso
33925f71e6
Add missing amp context if use_fp16 is enabled