We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent bffac75 commit 1d1eff1Copy full SHA for 1d1eff1
models/rank/naml/config_bigdata.yaml
@@ -17,16 +17,16 @@ runner:
17
train_reader_path: "NAMLDataReader" # importlib format
18
use_gpu: False
19
train_batch_size: 50
20
- epochs: 3
+ epochs: 2
21
print_interval: 10
22
#model_init_path: "output_model/0" # init model
23
model_save_path: "output_model_all"
24
infer_batch_size: 10
25
infer_reader_path: "NAMLDataReader" # importlib format
26
test_data_dir: "../../../datasets/MIND/data/test"
27
infer_load_path: "output_model_all"
28
- infer_start_epoch: 3
29
- infer_end_epoch: 4
+ infer_start_epoch: 2
+ infer_end_epoch: 3
30
31
# hyper parameters of user-defined network
32
hyper_parameters:
0 commit comments