comments and cleanup

huggingface · Dec 13, 2024 · 4918dcd · 4918dcd
1 parent ded96fb
commit 4918dcd
Showing 1 changed file with 14 additions and 14 deletions.
diff --git a/examples/contrastive-image-text/README.md b/examples/contrastive-image-text/README.md
@@ -132,17 +132,17 @@ PT_HPU_LAZY_MODE=0 PT_ENABLE_INT64_SUPPORT=1 \
 python3 ../gaudi_spawn.py --world_size 8 --use_mpi run_clip.py \
     --output_dir=/tmp/clip_roberta \
     --model_name_or_path=./clip-roberta \
-    --data_dir=./data \
-    --dataset_name="ydshieh/coco_dataset_script" \
-    --dataset_config_name="2017" \
-    --image_column="image_path" \
-    --caption_column="caption" \
-    --remove_unused_columns="False" \
+    --data_dir $PWD/data \
+    --dataset_name ydshieh/coco_dataset_script \
+    --dataset_config_name 2017 \
+    --image_column image_path \
+    --caption_column caption \
+    --remove_unused_columns=False \
     --do_train --do_eval \
     --mediapipe_dataloader \
     --per_device_train_batch_size="64" \
     --per_device_eval_batch_size="64" \
-    --learning_rate="5e-5" --warmup_steps="0" --weight_decay="0.1" \
+    --learning_rate="5e-5" --warmup_steps="0" --weight_decay 0.1 \
     --overwrite_output_dir \
     --use_habana \
     --use_lazy_mode=False \
@@ -170,17 +170,17 @@ PT_HPU_LAZY_MODE=0 PT_ENABLE_INT64_SUPPORT=1 \
 python3 ../gaudi_spawn.py --world_size 8 --use_deepspeed run_clip.py \
     --output_dir=/tmp/clip_roberta \
     --model_name_or_path=./clip-roberta \
-    --data_dir=./data \
-    --dataset_name="ydshieh/coco_dataset_script" \
-    --dataset_config_name="2017" \
-    --image_column="image_path" \
-    --caption_column="caption" \
-    --remove_unused_columns="False" \
+    --data_dir $PWD/data \
+    --dataset_name ydshieh/coco_dataset_script \
+    --dataset_config_name 2017 \
+    --image_column image_path \
+    --caption_column caption \
+    --remove_unused_columns=False \
     --do_train --do_eval \
     --mediapipe_dataloader \
     --per_device_train_batch_size="64" \
     --per_device_eval_batch_size="64" \
-    --learning_rate="5e-5" --warmup_steps="0" --weight_decay="0.1" \
+    --learning_rate="5e-5" --warmup_steps="0" --weight_decay 0.1 \
     --overwrite_output_dir \
     --use_habana \
     --use_lazy_mode=False \