Skip to content

Commit 25a2c9e

Browse files
committed
Add kim model
1 parent 915cdc8 commit 25a2c9e

File tree

5 files changed

+7
-9
lines changed

5 files changed

+7
-9
lines changed

models/match/kim/config.yml renamed to models/match/kim/config.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515
# global settings
1616

1717
runner:
18-
train_data_dir: "/data/sample_data"
18+
train_data_dir: "./data/sample_data"
1919
train_reader_path: "mind_reader" # importlib format
2020
use_gpu: False
2121
use_auc: False
File renamed without changes.

models/match/kim/readme.md

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,7 @@ os : windows/linux/macos
5555
本文提供了样例数据可以供您快速体验,在任意目录下均可执行。在kim模型目录的快速执行命令如下:
5656
```bash
5757
# 进入模型目录
58-
# cd models/match/match-pyramid # 在任意目录均可运行
58+
# cd models/match/kim # 在任意目录均可运行
5959
# 动态图训练
6060
python -u trainer.py -m config.yaml -o mode=train # 全量数据运行config_bigdata.yaml
6161
# 动态图预测
@@ -87,10 +87,10 @@ python -u infer.py -m config.yaml -o mode=test
8787
cd ../../../datasets/kim
8888
bash run.sh
8989
```
90-
3. 切回模型目录f
90+
3. 切回模型目录
9191
```bash
92-
python -u trainer.py -m config_bigdata.yml -o mode=train
93-
python -u infer.py -m config_bigdata.yml -o mode=test
92+
python -u trainer.py -m config_bigdata.yaml -o mode=train
93+
python -u infer.py -m config_bigdata.yaml -o mode=test
9494
```
9595

9696
## 进阶使用

models/match/kim/run.sh

Lines changed: 0 additions & 2 deletions
This file was deleted.

test_tipc/configs/kim/train_infer_python.txt

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ runner.test_data_dir:test_tipc/data/train
1313
runner.train_data_dir:../../../test_tipc/data/train
1414
##
1515
trainer:norm_train
16-
norm_train:-u ./models/match/kim/trainer.py -m ./models/match/kim/config.yml -o runner.print_interval=2 mode=train
16+
norm_train:-u ./models/match/kim/trainer.py -m ./models/match/kim/config.yaml -o runner.print_interval=2 mode=train
1717
quant_train:null
1818
fpgm_train:null
1919
distill_train:null
@@ -27,7 +27,7 @@ null:null
2727
===========================infer_params===========================
2828
runner.model_save_path:
2929
runner.model_init_path:
30-
norm_export:-u test_tipc/configs/kim/to_static.py -m ./models/match/kim/config.yml -o runner.CE=true
30+
norm_export:-u test_tipc/configs/kim/to_static.py -m ./models/match/kim/config.yaml -o runner.CE=true
3131
quant_export:null
3232
fpgm_export:null
3333
distill_export:null

0 commit comments

Comments
 (0)