Skip to content

Commit 31fe453

Browse files
committed
fix ncf readme
1 parent 7c4beab commit 31fe453

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

models/recall/ncf/config.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@ runner:
2424

2525
test_data_dir: "data/test"
2626
infer_reader_path: "movielens_reader" # importlib format
27-
infer_batch_size: 5
27+
infer_batch_size: 1
2828
infer_load_path: "output_model_ncf"
2929
infer_start_epoch: 2
3030
infer_end_epoch: 3

0 commit comments

Comments
 (0)