Update params.json
Browse files- params.json +4 -4
params.json
CHANGED
|
@@ -1,7 +1,7 @@
|
|
| 1 |
{
|
| 2 |
-
"train_dataset_size":
|
| 3 |
-
"validation_dataset_size":
|
| 4 |
-
"baseline_dataset_size":
|
| 5 |
"overfit_test": 0,
|
| 6 |
|
| 7 |
"num_nodes": 30,
|
|
@@ -26,7 +26,7 @@
|
|
| 26 |
"learning_rate": 0.0001,
|
| 27 |
"batch_size": 16,
|
| 28 |
"test_batch_size": 16,
|
| 29 |
-
"num_epochs":
|
| 30 |
|
| 31 |
"dataset_path": "https://huggingface.co/datasets/Cainiao-AI/LaDe/resolve/main/delivery/delivery_sh.csv"
|
| 32 |
}
|
|
|
|
| 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,
|
|
|
|
| 26 |
"learning_rate": 0.0001,
|
| 27 |
"batch_size": 16,
|
| 28 |
"test_batch_size": 16,
|
| 29 |
+
"num_epochs": 50,
|
| 30 |
|
| 31 |
"dataset_path": "https://huggingface.co/datasets/Cainiao-AI/LaDe/resolve/main/delivery/delivery_sh.csv"
|
| 32 |
}
|