From 8b66329c2da67db64873443e0a40fcffd0c4ae38 Mon Sep 17 00:00:00 2001 From: Sasha Kolpakov Date: Wed, 1 Nov 2023 16:38:24 +0100 Subject: [PATCH] Update cifar_100_EfficientNet.py --- cifar_100_EfficientNet.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/cifar_100_EfficientNet.py b/cifar_100_EfficientNet.py index 7a57fe6..7105636 100644 --- 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