Skip to content

Commit 23350c4

Browse files
authored
change batch_size to 1024
1 parent 79674c9 commit 23350c4

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

‎featureRec/run_criteo.sh‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,5 +4,5 @@ CUDA_VISIBLE_DEVICES=0 python -m autoint.train \
44
--is_save --has_residual \
55
--save_path ./models/Criteo/b3h2_64x64x64/ \
66
--field_size 39 --run_times 1 \
7-
--epoch 3 --batch_size 128 \
7+
--epoch 3 --batch_size 1024 \
88

0 commit comments

Comments
 (0)