StyleDistance
/

mstyledistance

Feature Extraction

sentence-transformers

datadreamer-0.35.0

sentence-similarity

text-embeddings-inference

Model card Files Files and versions

AjayP13 commited on Dec 1, 2024

Commit

02e88bc

·

verified ·

1 Parent(s): 6457cee

Pushed by DataDreamer

Update training_args.json

Files changed (1) hide show

training_args.json +4 -4

training_args.json CHANGED Viewed

@@ -6,11 +6,11 @@
     "do_predict": false,
     "evaluation_strategy": "steps",
     "prediction_loss_only": false,
-    "per_device_train_batch_size": 64,
-    "per_device_eval_batch_size": 64,
     "per_gpu_train_batch_size": null,
     "per_gpu_eval_batch_size": null,
-    "gradient_accumulation_steps": 2,
     "eval_accumulation_steps": 1,
     "eval_delay": 0,
     "learning_rate": 0.0001,
@@ -28,7 +28,7 @@
     "log_level": "passive",
     "log_level_replica": "warning",
     "log_on_each_node": true,
-    "logging_dir": "./output/train-styledistance-model/_checkpoints/runs/Nov29_20-48-36_nlpgpu10.seas.upenn.edu",
     "logging_strategy": "steps",
     "logging_first_step": false,
     "logging_steps": 10,

     "do_predict": false,
     "evaluation_strategy": "steps",
     "prediction_loss_only": false,
+    "per_device_train_batch_size": 96,
+    "per_device_eval_batch_size": 96,
     "per_gpu_train_batch_size": null,
     "per_gpu_eval_batch_size": null,
+    "gradient_accumulation_steps": 1,
     "eval_accumulation_steps": 1,
     "eval_delay": 0,
     "learning_rate": 0.0001,
     "log_level": "passive",
     "log_level_replica": "warning",
     "log_on_each_node": true,
+    "logging_dir": "./output/train-styledistance-model/_checkpoints/runs/Nov30_22-23-47_nlpgpu09.seas.upenn.edu",
     "logging_strategy": "steps",
     "logging_first_step": false,
     "logging_steps": 10,