Skip to content

Commit bffac75

Browse files
committed
doc change
1 parent 819c869 commit bffac75

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

models/rank/naml/config_bigdata.yaml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -17,16 +17,16 @@ runner:
1717
train_reader_path: "NAMLDataReader" # importlib format
1818
use_gpu: False
1919
train_batch_size: 50
20-
epochs: 10
20+
epochs: 3
2121
print_interval: 10
2222
#model_init_path: "output_model/0" # init model
2323
model_save_path: "output_model_all"
2424
infer_batch_size: 10
2525
infer_reader_path: "NAMLDataReader" # importlib format
2626
test_data_dir: "../../../datasets/MIND/data/test"
2727
infer_load_path: "output_model_all"
28-
infer_start_epoch: 0
29-
infer_end_epoch: 10
28+
infer_start_epoch: 3
29+
infer_end_epoch: 4
3030

3131
# hyper parameters of user-defined network
3232
hyper_parameters:

0 commit comments

Comments
 (0)