Skip to content

Commit 7fd7561

Browse files
Update cifar_100_EfficientNet.py
Try with 50 samples per category
1 parent dc75e62 commit 7fd7561

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

cifar_100_EfficientNet.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,7 @@ def subsample_train(X_train, y_train, num_samples):
6363

6464
# We take only 20 samples in each category
6565

66-
num_samples = 20
66+
num_samples = 50
6767
X_sub, y_sub = subsample_train(X_train, y_train, num_samples)
6868

6969
# Preparing tensors for the training set and labels

0 commit comments

Comments
 (0)