decrease num_steps of test cases

alibaba · Sep 19, 2023 · 5bbead1 · 5bbead1
1 parent 8672e47
commit 5bbead1
Show file tree

Hide file tree

Showing 24 changed files with 28 additions and 28 deletions.
diff --git a/easy_rec/python/test/train_eval_test.py b/easy_rec/python/test/train_eval_test.py
@@ -833,13 +833,13 @@ def test_batch_tfrecord_input(self):
     self.assertTrue(self._success)
 
   def test_autodis_embedding(self):
-    self._success = test_utils.test_distributed_train_eval(
+    self._success = test_utils.test_single_train_eval(
         'samples/model_config/deepfm_on_criteo_with_autodis.config',
         self._test_dir)
     self.assertTrue(self._success)
 
   def test_periodic_embedding(self):
-    self._success = test_utils.test_distributed_train_eval(
+    self._success = test_utils.test_single_train_eval(
         'samples/model_config/deepfm_on_criteo_with_periodic.config',
         self._test_dir)
     self.assertTrue(self._success)

diff --git a/samples/model_config/autoint_on_sequence_feature_taobao.config b/samples/model_config/autoint_on_sequence_feature_taobao.config
@@ -19,7 +19,7 @@ train_config {
   }
   save_checkpoints_steps: 100
   sync_replicas: True
-  num_steps: 2500
+  num_steps: 100
 }
 
 eval_config {

diff --git a/samples/model_config/autoint_on_taobao.config b/samples/model_config/autoint_on_taobao.config
@@ -19,7 +19,7 @@ train_config {
   }
   save_checkpoints_steps: 100
   sync_replicas: True
-  num_steps: 2500
+  num_steps: 100
 }
 
 eval_config {

diff --git a/samples/model_config/bst_on_taobao.config b/samples/model_config/bst_on_taobao.config
@@ -19,7 +19,7 @@ train_config {
   }
   save_checkpoints_steps: 100
   sync_replicas: True
-  num_steps: 2500
+  num_steps: 100
 }
 
 eval_config {

diff --git a/samples/model_config/custom_early_stop_on_taobao.config b/samples/model_config/custom_early_stop_on_taobao.config
@@ -19,7 +19,7 @@ train_config {
   }
   save_checkpoints_steps: 50
   sync_replicas: True
-  num_steps: 500
+  num_steps: 200
 }
 
 eval_config {

diff --git a/samples/model_config/dbmtl_backbone_on_taobao.config b/samples/model_config/dbmtl_backbone_on_taobao.config
@@ -16,7 +16,7 @@ train_config {
     }
     use_moving_average: false
   }
-  num_steps: 200
+  num_steps: 100
   sync_replicas: true
   save_checkpoints_steps: 100
   log_step_count_steps: 100

diff --git a/samples/model_config/dbmtl_mmoe_on_taobao.config b/samples/model_config/dbmtl_mmoe_on_taobao.config
@@ -16,7 +16,7 @@ train_config {
     }
     use_moving_average: false
   }
-  num_steps: 5000
+  num_steps: 100
   sync_replicas: true
   save_checkpoints_steps: 100
   log_step_count_steps: 100

diff --git a/samples/model_config/dbmtl_on_multi_numeric_boundary_allow_key_transform.config b/samples/model_config/dbmtl_on_multi_numeric_boundary_allow_key_transform.config
@@ -16,7 +16,7 @@ train_config {
     }
     use_moving_average: false
   }
-  num_steps: 1000
+  num_steps: 100
   sync_replicas: true
   save_checkpoints_steps: 100
   log_step_count_steps: 100

diff --git a/samples/model_config/dssm_on_sequence_feature_taobao.config b/samples/model_config/dssm_on_sequence_feature_taobao.config
@@ -19,7 +19,7 @@ train_config {
   }
   save_checkpoints_steps: 100
   sync_replicas: false
-  num_steps: 2500
+  num_steps: 100
 }
 
 eval_config {

diff --git a/samples/model_config/dssm_on_taobao.config b/samples/model_config/dssm_on_taobao.config
@@ -19,7 +19,7 @@ train_config {
   }
   save_checkpoints_steps: 100
   sync_replicas: false
-  num_steps: 2500
+  num_steps: 100
 }
 
 eval_config {

diff --git a/samples/model_config/dssm_reg_on_taobao.config b/samples/model_config/dssm_reg_on_taobao.config
@@ -19,7 +19,7 @@ train_config {
   }
   save_checkpoints_steps: 100
   sync_replicas: false
-  num_steps: 2500
+  num_steps: 100
 }
 
 eval_config {

diff --git a/samples/model_config/dssm_with_sample_weight.config b/samples/model_config/dssm_with_sample_weight.config
@@ -3,7 +3,7 @@ eval_input_path: "data/test/test_sample_weight.txt"
 model_dir: 'experiments/dssm_with_sample_weight/'
 
 train_config {
-  log_step_count_steps: 200
+  log_step_count_steps: 100
   optimizer_config: {
     adam_optimizer: {
       learning_rate: {
@@ -18,7 +18,7 @@ train_config {
     use_moving_average: false
   }
 
-  save_checkpoints_steps: 500
+  save_checkpoints_steps: 100
 }
 
 eval_config {

diff --git a/samples/model_config/esmm_distribute_eval_on_taobao.config b/samples/model_config/esmm_distribute_eval_on_taobao.config
@@ -19,7 +19,7 @@ train_config {
   }
   save_checkpoints_steps: 100
   sync_replicas: True
-  num_steps: 1000
+  num_steps: 100
 }
 
 eval_config {

diff --git a/samples/model_config/esmm_on_sequence_feature_taobao.config b/samples/model_config/esmm_on_sequence_feature_taobao.config
@@ -19,7 +19,7 @@ train_config {
   }
   save_checkpoints_steps: 100
   sync_replicas: True
-  num_steps: 5000
+  num_steps: 100
 }
 
 eval_config {

diff --git a/samples/model_config/esmm_on_taobao.config b/samples/model_config/esmm_on_taobao.config
@@ -19,7 +19,7 @@ train_config {
   }
   save_checkpoints_steps: 100
   sync_replicas: True
-  num_steps: 5000
+  num_steps: 100
 }
 
 eval_config {

diff --git a/samples/model_config/esmm_variational_dropout_on_taobao.config b/samples/model_config/esmm_variational_dropout_on_taobao.config
@@ -19,7 +19,7 @@ train_config {
   }
   save_checkpoints_steps: 100
   sync_replicas: True
-  num_steps: 5000
+  num_steps: 200
 }
 
 eval_config {

diff --git a/samples/model_config/fibinet_on_taobao.config b/samples/model_config/fibinet_on_taobao.config
@@ -19,7 +19,7 @@ train_config {
   }
   save_checkpoints_steps: 100
   sync_replicas: True
-  num_steps: 400
+  num_steps: 100
 }
 
 eval_config {

diff --git a/samples/model_config/metric_learning_on_taobao.config b/samples/model_config/metric_learning_on_taobao.config
@@ -19,7 +19,7 @@ train_config {
   }
   save_checkpoints_steps: 100
   sync_replicas: false
-  num_steps: 2500
+  num_steps: 100
 }
 
 eval_config {

diff --git a/samples/model_config/multi_tower_on_taobao.config b/samples/model_config/multi_tower_on_taobao.config
@@ -19,7 +19,7 @@ train_config {
   }
   save_checkpoints_steps: 100
   sync_replicas: True
-  num_steps: 200
+  num_steps: 100
 }
 
 eval_config {

diff --git a/samples/model_config/pdn_on_taobao.config b/samples/model_config/pdn_on_taobao.config
@@ -5,7 +5,7 @@ eval_input_path: "data/test/tb_data/taobao_pdn_fake_test_data"
 model_dir: "experiments/pdn_on_taobao"
 
 train_config {
-  log_step_count_steps: 5
+  log_step_count_steps: 50
   optimizer_config: {
     adam_optimizer: {
       learning_rate: {
@@ -19,8 +19,8 @@ train_config {
     }
     use_moving_average: false
   }
-  save_summary_steps: 5
-  num_steps: 2000
+  save_summary_steps: 50
+  num_steps: 100
 }
 
 eval_config {

diff --git a/samples/model_config/ple_on_sequence_feature_taobao.config b/samples/model_config/ple_on_sequence_feature_taobao.config
@@ -16,7 +16,7 @@ train_config {
     }
     use_moving_average: false
   }
-  num_steps: 200
+  num_steps: 100
   sync_replicas: true
   save_checkpoints_steps: 100
   log_step_count_steps: 100

diff --git a/samples/model_config/ple_on_taobao.config b/samples/model_config/ple_on_taobao.config
@@ -16,7 +16,7 @@ train_config {
     }
     use_moving_average: false
   }
-  num_steps: 200
+  num_steps: 100
   sync_replicas: true
   save_checkpoints_steps: 100
   log_step_count_steps: 100

diff --git a/samples/model_config/simple_multi_task_on_sequence_feature_taobao.config b/samples/model_config/simple_multi_task_on_sequence_feature_taobao.config
@@ -16,7 +16,7 @@ train_config {
     }
     use_moving_average: false
   }
-  num_steps: 5000
+  num_steps: 100
   sync_replicas: true
   save_checkpoints_steps: 100
   log_step_count_steps: 100

diff --git a/samples/model_config/simple_multi_task_on_taobao.config b/samples/model_config/simple_multi_task_on_taobao.config
@@ -16,7 +16,7 @@ train_config {
     }
     use_moving_average: false
   }
-  num_steps: 5000
+  num_steps: 100
   sync_replicas: true
   save_checkpoints_steps: 100
   log_step_count_steps: 100