Štěpán Záliš

Sharing definition of the **model**: ``` model = tf.keras.models.Sequential() model.add(base_model) model.add(tf.keras.layers.Dropout(0.5)) model.add(tf.keras.layers.Dense( units=NUM_CLASSES, activation=tf.keras.activations.softmax, kernel_regularizer=tf.keras.regularizers.l2(l=0.01) )) ``` where **base_model** is Keras default model: ``` base_model = tf.keras.applications.MobileNetV2( input_shape=INPUT_IMG_SHAPE, include_top=False, weights='imagenet',...

I guess it's coming from dataset itself.

Alright, thanks a lot for solving it! 🙏