a-ragab-h-m commited on
Commit
59a13f9
·
verified ·
1 Parent(s): 89ab782

Update params.json

Browse files
Files changed (1) hide show
  1. params.json +6 -6
params.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
- "train_dataset_size": 5000, // 50000
3
- "validation_dataset_size": 1000, // 10000
4
- "baseline_dataset_size": 1000, // 10000
5
  "overfit_test": 0,
6
 
7
  "num_nodes": 30,
@@ -23,9 +23,9 @@
23
  "run_tests": true,
24
  "save_results": true,
25
  "learning_rate": 0.0001,
26
- "batch_size": 16, // 64
27
- "test_batch_size": 16, // 64
28
- "num_epochs": 50, // 1500
29
 
30
  "dataset_path": "https://huggingface.co/datasets/Cainiao-AI/LaDe/resolve/main/delivery/delivery_sh.csv"
31
  }
 
1
  {
2
+ "train_dataset_size": 5000,
3
+ "validation_dataset_size": 1000,
4
+ "baseline_dataset_size": 1000,
5
  "overfit_test": 0,
6
 
7
  "num_nodes": 30,
 
23
  "run_tests": true,
24
  "save_results": true,
25
  "learning_rate": 0.0001,
26
+ "batch_size": 16,
27
+ "test_batch_size": 16,
28
+ "num_epochs": 50,
29
 
30
  "dataset_path": "https://huggingface.co/datasets/Cainiao-AI/LaDe/resolve/main/delivery/delivery_sh.csv"
31
  }