Skip to content

Commit fa3b725

Browse files
authored
Update config.yaml
modify infer batch size
1 parent 851cb3e commit fa3b725

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

models/rank/xdeepfm/config.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ runner:
2828
infer_batch_size: 5
2929
infer_load_path: "output_model_deepfm"
3030
infer_start_epoch: 0
31-
infer_end_epoch: 3
31+
infer_end_epoch: 1
3232

3333

3434
# hyper parameters of user-defined network

0 commit comments

Comments
 (0)