Skip to content

Commit 68ba6a5

Browse files
committed
fix dlrm ci
1 parent eba457d commit 68ba6a5

File tree

3 files changed

+4
-46
lines changed

3 files changed

+4
-46
lines changed

models/rank/dlrm/config.yaml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ runner:
3838
thread_num: 1
3939

4040
# use inference save model
41-
inference: True # 静态图训练时保存为inference model
41+
inference: False # 静态图训练时保存为inference model
4242
save_inference_feed_varnames: ["C1","C2","C3","C4","C5","C6","C7","C8","C9","C10","C11","C12","C13","C14","C15","C16","C17","C18","C19","C20","C21","C22","C23","C24","C25","C26","dense_input"] # inference model 的feed参数的名字
4343
save_inference_fetch_varnames: ["sigmoid_0.tmp_0"] # inference model 的fetch参数的名字
4444

@@ -55,5 +55,5 @@ hyper_parameters:
5555
bot_layer_sizes: [512, 256, 64, 16]
5656
sparse_feature_number: 1000001
5757
sparse_feature_dim: 16
58-
top_layer_sizes: [512, 256, 1]
58+
top_layer_sizes: [512, 256, 2]
5959
num_field: 26

models/recall/ensfm/config_bigdata.yaml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ runner:
1616
train_data_dir: "../../../datasets/ml-1m_ensfm/data/ml-1m-ensfm"
1717
train_reader_path: "movielens_reader" # importlib format
1818
train_batch_size: 512
19-
model_save_path: "output_model_ensfm"
19+
model_save_path: "output_model_ensfm_all"
2020
mode: "train"
2121
use_gpu: True
2222
epochs: 501
@@ -25,7 +25,7 @@ runner:
2525
test_data_dir: "../../../datasets/ml-1m_ensfm/data/ml-1m-ensfm"
2626
infer_reader_path: "movielens_reader" # importlib format
2727
infer_batch_size: 512
28-
infer_load_path: "output_model_ensfm"
28+
infer_load_path: "output_model_ensfm_all"
2929
infer_start_epoch: 100
3030
infer_end_epoch: 501
3131

models/recall/tisas/config_test.yaml

Lines changed: 0 additions & 42 deletions
This file was deleted.

0 commit comments

Comments
 (0)