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 819c869 commit bffac75Copy full SHA for bffac75
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: 10
+ epochs: 3
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: 0
29
- infer_end_epoch: 10
+ infer_start_epoch: 3
+ infer_end_epoch: 4
30
31
# hyper parameters of user-defined network
32
hyper_parameters:
0 commit comments