mirror of
https://github.com/huggingface/trl.git
synced 2025-10-20 18:43:52 +08:00
LayerSkip Training Recipe
Implements the training recipe as described in the LayerSkip paper.
Run training
cd scripts
python layer_skip_sft.py
Run benchmark
cd scripts
python benchmark_layer_skip.py