Skip to content

Commit 5a7d0f0

Browse files
committed
autofis fix config
1 parent 80e4d93 commit 5a7d0f0

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

models/rank/autofis/config.yaml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,11 +23,11 @@ runner:
2323
epochs: 1
2424
print_interval: 1
2525
#model_init_path: "output_model/0" # init model
26-
model_save_path: "output_model_demo_autodeepfm"
26+
model_save_path: "output_model_autofis"
2727
test_data_dir: "data/sample_data/test"
2828
infer_reader_path: "criteo_reader" # importlib format
2929
infer_batch_size: 2
30-
infer_load_path: "output_model_demo_autodeepfm"
30+
infer_load_path: "output_model_autofis"
3131
infer_start_epoch: 0
3232
infer_end_epoch: 1
3333

models/rank/autofis/config_bigdata.yaml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -23,11 +23,11 @@ runner:
2323
epochs: 1
2424
print_interval: 50
2525
#model_init_path: "output_model/0" # init model
26-
model_save_path: "output_model_all_autodeepfm"
26+
model_save_path: "output_model_autofis_all"
2727
test_data_dir: "../../../datasets/criteo_autofis/data/whole_data/test"
2828
infer_reader_path: "criteo_reader" # importlib format
29-
infer_batch_size: 2000
30-
infer_load_path: "output_model_all_autodeepfm"
29+
infer_batch_size: 200
30+
infer_load_path: "output_model_autofis_all"
3131
infer_start_epoch: 0
3232
infer_end_epoch: 1
3333

0 commit comments

Comments
 (0)