Активировать отсев в предварительно обученной модели VGG16

Question

Я использую Tensorflow 2.0 и предварительно обученную модель VGG16 и хочу активировать отсев во время прогнозирования. До сих пор я безуспешно пробовал следующее:

model = tf.keras.applications.VGG16(input_shape=(224, 224, 3), weights='imagenet', is_training=True)

model = tf.keras.applications.VGG16(input_shape=(224, 224, 3), weights='imagenet', dropout_rate=0.5)

Однако ни один из этих подходов не сработал. Как включить отсев на этапе прогнозирования?

Answer 1

Архитектура VGG16 не содержит выпадающий слой по умолчанию. Вам необходимо вставить выпадающий слой в модель.

Вот пост, который я нашел полезным для решения этой проблемы: Добавление выпадающих слоев между плотными слоями с предварительной подготовкой в ​​кератах