Update cifar_100_EfficientNet.py

sashakolpakov · web-flow · commit 8b66329c2da6 · 2023-11-01T16:38:24.000+01:00
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

Original file line number	Diff line number	Diff line change
`@@ -61,9 +61,9 @@ def subsample_train(X_train, y_train, num_samples):`
`61`	`61`	`#`
`62`	`62`	`return X_sub[ind], y_sub[ind]`
`63`	`63`
`64`		`-# We take only 50 samples in each category`
	`64`	`+# We take only 20 samples in each category`
`65`	`65`
`66`		`-num_samples = 50`
	`66`	`+num_samples = 20`
`67`	`67`	`X_sub, y_sub = subsample_train(X_train, y_train, num_samples)`
`68`	`68`
`69`	`69`	`# Preparing tensors for the training set and labels`