Skip to content

Commit 89f532f

Browse files
authored
Merge pull request #749 from wangzhen38/tipc_update
tipc-update
2 parents 5512ac4 + f6e1a44 commit 89f532f

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

test_tipc/configs/tisas/paddle_infer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -107,7 +107,7 @@ def create_data_loader(args):
107107
sys.path.append(reader_path)
108108
# sys.path.append(os.path.abspath("."))
109109
reader_class = import_module(reader_file)
110-
config = load_yaml('./models/recall/tisas/config_test.yaml')
110+
config = load_yaml('./models/recall/tisas/config.yaml')
111111
dataset = reader_class.RecDataset(file_list, config=config)
112112
loader = DataLoader(
113113
dataset, batch_size=batchsize, places=place, drop_last=True)

test_tipc/configs/tisas/train_infer_python.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
===========================train_params===========================
22
model_name:tisas
3-
python:python3
3+
python:python3.7
44
gpu_list:0|-1
55
runner.use_gpu:True|False
66
auto_cast:False

0 commit comments

Comments
 (0)