Skip to content

Commit 48b0de7

Browse files
committed
aitm fix
1 parent 73c2277 commit 48b0de7

File tree

9 files changed

+2
-2
lines changed

9 files changed

+2
-2
lines changed
File renamed without changes.
File renamed without changes.

models/rank/aitm/config.yaml renamed to models/multitask/aitm/config.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@ runner:
2424
#model_init_path: "output_model/0" # init model
2525
model_save_path: "output_model_aitm_test/"
2626
test_data_dir: "./data/sample_data/test"
27-
infer_reader_path: "reader" # importlib format
27+
infer_reader_path: "aitm_reader" # importlib format
2828
infer_batch_size: 10
2929
infer_load_path: "output_model_aitm_test/"
3030
infer_start_epoch: 0

models/rank/aitm/config_bigdata.yaml renamed to models/multitask/aitm/config_bigdata.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@ runner:
2424
#model_init_path: "output_model/0" # init model
2525
model_save_path: "output_model_aitm_all/"
2626
test_data_dir: "../../../datasets/ali-cpp_aitm/whole_data/test"
27-
infer_reader_path: "reader" # importlib format
27+
infer_reader_path: "aitm_reader" # importlib format
2828
infer_batch_size: 2000
2929
infer_load_path: "output_model_aitm_all/"
3030
infer_start_epoch: 0
File renamed without changes.
File renamed without changes.
File renamed without changes.

0 commit comments

Comments
 (0)