Update cifar_100_EfficientNet.py

david-thrower · Nov 1, 2023 · 8b66329 · 8b66329
1 parent 29ec2ae
commit 8b66329
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/cifar_100_EfficientNet.py b/cifar_100_EfficientNet.py
@@ -61,9 +61,9 @@ def subsample_train(X_train, y_train, num_samples):
     #
     return X_sub[ind], y_sub[ind]
 
-# We take only 50 samples in each category
+# We take only 20 samples in each category
 
-num_samples = 50
+num_samples  = 20
 X_sub, y_sub = subsample_train(X_train, y_train, num_samples)
 
 # Preparing tensors for the training set and labels