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 7c4beab commit 31fe453Copy full SHA for 31fe453
models/recall/ncf/config.yaml
@@ -24,7 +24,7 @@ runner:
24
25
test_data_dir: "data/test"
26
infer_reader_path: "movielens_reader" # importlib format
27
- infer_batch_size: 5
+ infer_batch_size: 1
28
infer_load_path: "output_model_ncf"
29
infer_start_epoch: 2
30
infer_end_epoch: 3
0 commit comments