Skip to content

Commit

Permalink
fix bug of wrong demo model configs (#508)
Browse files Browse the repository at this point in the history
* fix bug of autoint model demo config
  • Loading branch information
yangxudong authored Dec 25, 2024
1 parent d965fb7 commit b07889a
Show file tree
Hide file tree
Showing 12 changed files with 3 additions and 36 deletions.
4 changes: 2 additions & 2 deletions examples/configs/autoint_on_movielens.config
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@ eval_input_path: "examples/data/movielens_1m/movies_test_data"
model_dir: "examples/ckpt/autoint_on_movieslen_ckpt"

train_config {
log_step_count_steps: 1000
log_step_count_steps: 100
optimizer_config: {
adam_optimizer: {
learning_rate: {
Expand All @@ -17,7 +17,7 @@ train_config {
}
use_moving_average: false
}
save_checkpoints_steps: 100
save_checkpoints_steps: 1000
sync_replicas: True
num_steps: 2500
}
Expand Down
5 changes: 1 addition & 4 deletions examples/configs/dcn_backbone_on_movielens.config
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
train_input_path: "examples/data/movielens_1m/movies_train_data"
eval_input_path: "examples/data/movielens_1m/movies_test_data"
model_dir: "examples/ckpt/dcn_on_movieslen"
model_dir: "examples/ckpt/dcn_backbone_on_movieslen"

train_config {
log_step_count_steps: 100
Expand Down Expand Up @@ -199,6 +199,3 @@ model_config: {
}
embedding_regularization: 1e-4
}
export_config {
multi_placeholder: false
}
3 changes: 0 additions & 3 deletions examples/configs/dcn_on_movielens.config
Original file line number Diff line number Diff line change
Expand Up @@ -180,6 +180,3 @@ model_config: {
}
embedding_regularization: 1e-4
}
export_config {
multi_placeholder: false
}
3 changes: 0 additions & 3 deletions examples/configs/deepfm_backbone_on_movielens.config
Original file line number Diff line number Diff line change
Expand Up @@ -241,6 +241,3 @@ model_config: {
}
embedding_regularization: 1e-4
}
export_config {
multi_placeholder: false
}
3 changes: 0 additions & 3 deletions examples/configs/deepfm_on_movielens.config
Original file line number Diff line number Diff line change
Expand Up @@ -182,6 +182,3 @@ model_config: {
}
embedding_regularization: 1e-4
}
export_config {
multi_placeholder: false
}
3 changes: 0 additions & 3 deletions examples/configs/dssm_on_books.config
Original file line number Diff line number Diff line change
Expand Up @@ -112,6 +112,3 @@ model_config:{
}
embedding_regularization: 5e-5
}

export_config {
}
3 changes: 0 additions & 3 deletions examples/configs/dssm_on_books_negative_sample.config
Original file line number Diff line number Diff line change
Expand Up @@ -127,6 +127,3 @@ model_config:{
loss_type: SOFTMAX_CROSS_ENTROPY
embedding_regularization: 5e-6
}

export_config {
}
3 changes: 0 additions & 3 deletions examples/configs/dssm_senet_on_taobao.config
Original file line number Diff line number Diff line change
Expand Up @@ -281,6 +281,3 @@ model_config:{
}
embedding_regularization: 5e-5
}

export_config {
}
3 changes: 0 additions & 3 deletions examples/configs/fibinet_on_movielens.config
Original file line number Diff line number Diff line change
Expand Up @@ -200,6 +200,3 @@ model_config: {
}
embedding_regularization: 1e-4
}
export_config {
multi_placeholder: false
}
3 changes: 0 additions & 3 deletions examples/configs/masknet_on_movielens.config
Original file line number Diff line number Diff line change
Expand Up @@ -196,6 +196,3 @@ model_config: {
}
embedding_regularization: 1e-4
}
export_config {
multi_placeholder: false
}
3 changes: 0 additions & 3 deletions examples/configs/mind_on_books.config
Original file line number Diff line number Diff line change
Expand Up @@ -111,6 +111,3 @@ model_config:{
}
embedding_regularization: 5e-5
}

export_config {
}
3 changes: 0 additions & 3 deletions examples/configs/mind_on_books_negative_sample.config
Original file line number Diff line number Diff line change
Expand Up @@ -128,6 +128,3 @@ model_config:{
embedding_regularization: 5e-5
loss_type: SOFTMAX_CROSS_ENTROPY
}

export_config {
}

0 comments on commit b07889a

Please sign in to comment.