mirror of
https://github.com/wassname/GENIES.git
synced 2026-08-13 11:50:49 +08:00
Initial commit
This commit is contained in:
@@ -0,0 +1,8 @@
|
||||
|
||||
export TRAINKWARGS='{"num_train_steps": 20, "per_device_train_batch_size": 32}'
|
||||
python src/api/train.py \
|
||||
--model_dir models/pythia-410m \
|
||||
--intervention_dir src/interventions/lora_fine_tune \
|
||||
--train_distribution distributions/alpaca_mmlu \
|
||||
--train_kwargs "$TRAINKWARGS" \
|
||||
--retrain
|
||||
Reference in New Issue
Block a user