fix aitm

Author: yinhaofeng

models/multitask/aitm/readme.md

@@ -54,7 +54,7 @@ os : windows/linux/macos
 本文提供了样例数据可以供您快速体验，在任意目录下均可执行。在aitm模型目录的快速执行命令如下： 
 ```bash
 # 进入模型目录
-# cd models/rank/aitm # 在任意目录均可运行
+# cd models/multitask/aitm # 在任意目录均可运行
 # 动态图训练
 python -u ../../../tools/trainer.py -m config.yaml
 
@@ -68,7 +68,7 @@ python -u ../../../tools/infer.py -m config.yaml
 | :------| :------ | :------ | :------ | :------| :------ | 
 | aitm | 0.6186 |0.6525 | 2000 | 6| 约3小时 |
 
-1. 确认您当前所在目录为PaddleRec/models/rank/aitm
+1. 确认您当前所在目录为PaddleRec/models/multitask/aitm
 2. 进入Paddlerec/datasets/ali-cpp_aitm
 3. 执行命令运行全量数据
 

test_tipc/configs/aitm/train_infer_python.txt

@@ -13,7 +13,7 @@ runner.test_data_dir:test_tipc/data/infer
 runner.train_data_dir:../../../test_tipc/data/train
 ##
 trainer:norm_train
-norm_train:-u tools/trainer.py -m ./models/rank/aitm/config.yaml -o runner.print_interval=2
+norm_train:-u tools/trainer.py -m ./models/multitask/aitm/config.yaml -o runner.print_interval=2
 quant_train:null
 fpgm_train:null
 distill_train:null
@@ -27,7 +27,7 @@ null:null
 ===========================infer_params===========================
 runner.model_save_path:
 runner.model_init_path:
-norm_export:-u test_tipc/configs/aitm/to_static.py -m ./models/rank/aitm/config.yaml -o runner.CE=true
+norm_export:-u test_tipc/configs/aitm/to_static.py -m ./models/multitask/aitm/config.yaml -o runner.CE=true
 quant_export:null
 fpgm_export:null
 distill_export:null
@@ -37,7 +37,7 @@ null:null
 infer_model:test_tipc/save_aitm_model
 infer_export:null
 infer_quant:False
-inference:-u test_tipc/configs/aitm/paddle_infer.py --model_name=aitm --reader_file=./models/rank/aitm/reader.py
+inference:-u test_tipc/configs/aitm/paddle_infer.py --model_name=aitm --reader_file=./models/multitask/aitm/aitm_reader.py
 --use_gpu:True|False
 --enable_mkldnn:True|False
 --cpu_threads:1|6

test_tipc/prepare.sh

@@ -306,8 +306,8 @@ elif [ ${model_name} == "aitm" ]; then
     mkdir -p ./test_tipc/data/train
     mkdir -p ./test_tipc/data/infer
     if [ ${MODE} = "lite_train_lite_infer" ];then
-        cp -r ./models/rank/aitm/data/sample_data/train/* ./test_tipc/data/train
-        cp -r ./models/rank/aitm/data/sample_data/test/* ./test_tipc/data/infer
+        cp -r ./models/multitask/aitm/data/sample_data/train/* ./test_tipc/data/train
+        cp -r ./models/multitask/aitm/data/sample_data/test/* ./test_tipc/data/infer
         echo "demo data ready"
     elif [ ${MODE} = "whole_train_whole_infer" ];then
         cd ./datasets/ali-cpp_aitm