Skip to content

Commit 1d1eff1

Browse files
committed
decrease epoch
1 parent bffac75 commit 1d1eff1

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: 3
20+
epochs: 2
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: 3
29-
infer_end_epoch: 4
28+
infer_start_epoch: 2
29+
infer_end_epoch: 3
3030

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

0 commit comments

Comments
 (0)