Pushed by DataDreamer

Update training_args.json

training_args.json

@@ -13,13 +13,13 @@
     "gradient_accumulation_steps": 1,
     "eval_accumulation_steps": 1,
     "eval_delay": 0,
-    "learning_rate": 0.001,
+    "learning_rate": 0.0001,
     "weight_decay": 0.01,
     "adam_beta1": 0.9,
     "adam_beta2": 0.999,
     "adam_epsilon": 1e-08,
     "max_grad_norm": 1.0,
-    "num_train_epochs": 1,
+    "num_train_epochs": 10,
     "max_steps": -1,
     "lr_scheduler_type": "linear",
     "lr_scheduler_kwargs": {},
@@ -28,7 +28,7 @@
     "log_level": "passive",
     "log_level_replica": "warning",
     "log_on_each_node": true,
-    "logging_dir": "./output/train-styledistance-model/_checkpoints/runs/Jul17_08-19-41_nlpgpu04.seas.upenn.edu",
+    "logging_dir": "./output/train-styledistance-model/_checkpoints/runs/Jul17_08-37-06_nlpgpu04.seas.upenn.edu",
     "logging_strategy": "steps",
     "logging_first_step": false,
     "logging_steps": 1,