Update params.json

params.json

@@ -1,7 +1,7 @@
 {
-  "train_dataset_size": 50000,
-  "validation_dataset_size": 10000,
-  "baseline_dataset_size": 10000,
+  "train_dataset_size": 5000, // 50000
+  "validation_dataset_size": 1000,  // 10000
+  "baseline_dataset_size": 1000,   // 10000
   "overfit_test": 0,
 
   "num_nodes": 30,
@@ -23,9 +23,9 @@
   "run_tests": true,
   "save_results": true,
   "learning_rate": 0.0001,
-  "batch_size": 64,
-  "test_batch_size": 64,
-  "num_epochs": 1500,
+  "batch_size": 16,       // 64
+  "test_batch_size": 16,   // 64
+  "num_epochs": 50,     // 1500
 
   "dataset_path": "https://huggingface.co/datasets/Cainiao-AI/LaDe/resolve/main/delivery/delivery_sh.csv"
 }