Skip to content

Commit ccb22f0

Browse files
sutaakaropenshift-merge-bot[bot]
authored andcommitted
Provide correct log directory for kfto_training_test
1 parent c2c5ea6 commit ccb22f0

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/kfto/kfto_training_test.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -193,7 +193,7 @@ func createKFTOPyTorchJob(test Test, namespace string, config corev1.ConfigMap,
193193
--model_dir /tmp/model/bloom-560m \
194194
--dataset_file /tmp/all_datasets/alpaca_data_tenth.json \
195195
--transformer_type AutoModelForCausalLM \
196-
--training_parameters '{"output_dir": "/mnt/output", "per_device_train_batch_size": 8, "num_train_epochs": 3, "logging_dir": "/logs", "eval_strategy": "epoch", "save_strategy": "no"}' \
196+
--training_parameters '{"output_dir": "/mnt/output", "per_device_train_batch_size": 8, "num_train_epochs": 3, "logging_dir": "/tmp/logs", "eval_strategy": "epoch", "save_strategy": "no"}' \
197197
--lora_config '{"r": 4, "lora_alpha": 16, "lora_dropout": 0.1, "bias": "none"}'`,
198198
},
199199
Env: []corev1.EnvVar{

0 commit comments

Comments
 (0)