Merge pull request #767 from yinhaofeng/aitm

fix aitm

Author: frankwhzhang

datasets/Ali_Display_Ad_Click_DSIN/run.sh

@@ -1,6 +1,6 @@
 mkdir big_train
 mkdir big_test
-wget -O model_input.tar.gz https://bj.bcebos.com/v1/ai-studio-online/53e61a9bcfc54e0581044883d0f876d9841cb4d0a68848f1a1d568a84591da6f?responseContentDisposition=attachment%3B%20filename%3Dmodel_input.tar.gz&authorization=bce-auth-v1%2F0ef6765c1e494918bc0d4c3ca3e5c6d1%2F2022-04-21T01%3A43%3A00Z%2F-1%2F%2F665a728726f0569e1ef9dd423adfa40a2a5e798f86a8d5d68804a2f21cc03624
+wget https://paddlerec.bj.bcebos.com/datasets/Ali_Display_Ad_Click/model_input.tar.gz
 tar -zxvf model_input.tar.gz
 mv model_input/test_feat_input.pkl big_test/
 mv model_input/test_label.pkl big_test/

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
 
@@ -66,9 +66,9 @@ python -u ../../../tools/infer.py -m config.yaml
 在全量数据下模型的指标如下：  
 | 模型 | click auc | purchase auc |batch_size | epoch_num| Time of each epoch |
 | :------| :------ | :------ | :------ | :------| :------ | 
-| aitm | 0.6186 |0.6525 | 2000 | 6| 约3小时 |
+| aitm | 0.6130 |0.6166 | 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

@@ -315,8 +315,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