Skip to content

Commit e2c6829

Browse files
committed
path change
1 parent 277c57e commit e2c6829

File tree

2 files changed

+4
-6
lines changed

2 files changed

+4
-6
lines changed

models/rank/naml/config.yaml

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -17,14 +17,14 @@ runner:
1717
train_reader_path: "NAMLDataReader" # importlib format
1818
use_gpu: False
1919
train_batch_size: 10
20-
epochs: 2000
20+
epochs: 20
2121
print_interval: 2
2222
#model_init_path: "output_model/0" # init model
23-
model_save_path: "output_model"
23+
model_save_path: "output_model_all"
2424
infer_batch_size: 2
2525
infer_reader_path: "NAMLDataReader" # importlib format
2626
test_data_dir: "data/sample_data/train"
27-
infer_load_path: "output_model"
27+
infer_load_path: "output_model_all"
2828
infer_start_epoch: 0
2929
infer_end_epoch: 3
3030

@@ -45,4 +45,3 @@ hyper_parameters:
4545
sub_category_size: 10
4646
category_dimension: 32
4747
word_dict_size: 101
48-
fc_sizes: [512, 256, 128, 32]

models/rank/naml/config_bigdata.yaml

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ runner:
2626
test_data_dir: "../../../datasets/MIND/data/test"
2727
infer_load_path: "output_model"
2828
infer_start_epoch: 0
29-
infer_end_epoch: 1
29+
infer_end_epoch: 2
3030

3131
# hyper parameters of user-defined network
3232
hyper_parameters:
@@ -45,4 +45,3 @@ hyper_parameters:
4545
sub_category_size: 260
4646
category_dimension: 32
4747
word_dict_size: 19200
48-
fc_sizes: [512, 256, 128, 32]

0 commit comments

Comments
 (0)