File tree Expand file tree Collapse file tree
ci/benchmarks/partial-conv Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -44,19 +44,19 @@ script_args:
4444 precision : fp8
4545 products :
4646 - variant : finetune
47- lora_enabled : false
47+ lora_enabled : " "
4848 task : finetune_from_ckpt
4949 experiment_name : evo2-finetune
5050 - variant : lora_finetune
51- lora_enabled : true
51+ lora_enabled : " --lora-finetune "
5252 task : lora_finetune_from_ckpt
5353 experiment_name : evo2-lora-finetune
5454script : |-
5555 WANDB_API_KEY=$BIONEMO_WANDB_API_KEY train_${model} \
5656 -d ${data_base_path}/${dataset_config} \
5757 --dataset-dir=${data_base_path}/${dataset_dir} \
5858 --ckpt-dir=${data_base_path}/${restore_from_checkpoint_path} \
59- ${ if lora_enabled then "--lora-finetune" else "" } \
59+ ${lora_enabled} \
6060 --model-size=${model_size} \
6161 --max-steps=${max_steps} \
6262 --experiment-name=${experiment_name}_${batch_size}bs_${nodes}node_${gpus}gpu_${max_steps}s \
You can’t perform that action at this time.
0 commit comments