Add kim model

hoszeman · Jun 8, 2022 · 25a2c9e · 25a2c9e
1 parent 915cdc8
commit 25a2c9e
Show file tree

Hide file tree

Showing 5 changed files with 7 additions and 9 deletions.
diff --git a/models/match/kim/config.yml → models/match/kim/config.yaml b/models/match/kim/config.yml → models/match/kim/config.yaml
@@ -15,7 +15,7 @@
 # global settings
 
 runner:
-  train_data_dir: "/data/sample_data"
+  train_data_dir: "./data/sample_data"
   train_reader_path: "mind_reader" # importlib format
   use_gpu: False
   use_auc: False

diff --git a/models/match/kim/config_bigdata.yml → models/match/kim/config_bigdata.yaml b/models/match/kim/config_bigdata.yml → models/match/kim/config_bigdata.yaml
diff --git a/models/match/kim/readme.md b/models/match/kim/readme.md
@@ -55,7 +55,7 @@ os : windows/linux/macos
 本文提供了样例数据可以供您快速体验，在任意目录下均可执行。在kim模型目录的快速执行命令如下： 
 ```bash
 # 进入模型目录
-# cd models/match/match-pyramid # 在任意目录均可运行
+# cd models/match/kim # 在任意目录均可运行
 # 动态图训练
 python -u trainer.py -m config.yaml -o mode=train # 全量数据运行config_bigdata.yaml 
 # 动态图预测
@@ -87,10 +87,10 @@ python -u infer.py -m config.yaml -o mode=test
 cd ../../../datasets/kim
 bash run.sh
 ```
-3. 切回模型目录f
+3. 切回模型目录
 ```bash
-python -u trainer.py -m config_bigdata.yml -o mode=train
-python -u infer.py -m config_bigdata.yml -o mode=test
+python -u trainer.py -m config_bigdata.yaml -o mode=train
+python -u infer.py -m config_bigdata.yaml -o mode=test
 ```
 
 ## 进阶使用

diff --git a/models/match/kim/run.sh b/models/match/kim/run.sh
diff --git a/test_tipc/configs/kim/train_infer_python.txt b/test_tipc/configs/kim/train_infer_python.txt
@@ -13,7 +13,7 @@ runner.test_data_dir:test_tipc/data/train
 runner.train_data_dir:../../../test_tipc/data/train
 ##
 trainer:norm_train
-norm_train:-u ./models/match/kim/trainer.py -m ./models/match/kim/config.yml -o runner.print_interval=2 mode=train
+norm_train:-u ./models/match/kim/trainer.py -m ./models/match/kim/config.yaml -o runner.print_interval=2 mode=train
 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/kim/to_static.py -m ./models/match/kim/config.yml -o runner.CE=true
+norm_export:-u test_tipc/configs/kim/to_static.py -m ./models/match/kim/config.yaml -o runner.CE=true
 quant_export:null
 fpgm_export:null
 distill_export:null