Skip to content

Commit c66cad8

Browse files
author
tdeboissiere
committed
Change batch limitation
1 parent a122bd3 commit c66cad8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

DenseRecNet/run_cifar10.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -128,7 +128,7 @@ def run_cifar10(batch_size,
128128
l_train_loss = []
129129
start = time.time()
130130

131-
for batch_idx in arr_splits[:10]:
131+
for batch_idx in arr_splits:
132132

133133
X_batch, Y_batch = X_train[batch_idx], Y_train[batch_idx]
134134
train_logloss, train_acc = model.train_on_batch(X_batch, Y_batch)

0 commit comments

Comments
 (0)