Utilizando pesas pre-entrenadas ResNet50 Estoy tratando de construir un clasificador. La base de código está completamente implementada en la API de Tensorflow de alto nivel de Keras. El código completo se publica en el siguiente enlace de GitHub.

Código fuente: Clasificación usando la arquitectura RestNet50

El tamaño del archivo del modelo pre-entrenado es 94.7mb.

Cargué el archivo pre-entrenado

new_model = Sequential()

new_model.add(ResNet50(include_top=False,
                pooling='avg',
                weights=resnet_weight_paths))

y se ajusta al modelo

train_generator = data_generator.flow_from_directory(
    'path_to_the_training_set',
    target_size = (IMG_SIZE,IMG_SIZE),
    batch_size = 12,
    class_mode = 'categorical'
    )

validation_generator = data_generator.flow_from_directory(
    'path_to_the_validation_set',
    target_size = (IMG_SIZE,IMG_SIZE),
    class_mode = 'categorical'
    )

#compile the model

new_model.fit_generator(
    train_generator,
    steps_per_epoch = 3,
    validation_data = validation_generator,
    validation_steps = 1
)

y en el conjunto de datos de Entrenamiento, tengo dos carpetas para perros y gatos, cada una con casi 10,000 imágenes. Cuando compilé el script, recibo el siguiente error

Epoch 1/1 2018-05-12 13: 04: 45.847298: W tensorflow / core / framework / allocator.cc: 101] La asignación de 38535168 excede el 10% de la memoria del sistema. 2018-05-12 13: 04: 46.845021: W tensorflow / core / framework / allocator.cc: 101] La asignación de 37171200 excede el 10% de la memoria del sistema. 2018-05-12 13: 04: 47.552176: W tensorflow / core / framework / allocator.cc: 101] La asignación de 37171200 excede el 10% de la memoria del sistema. 2018-05-12 13: 04: 48.199240: W tensorflow / core / framework / allocator.cc: 101] La asignación de 37171200 excede el 10% de la memoria del sistema. 2018-05-12 13: 04: 48.918930: W tensorflow / core / framework / allocator.cc: 101] La asignación de 37171200 excede el 10% de la memoria del sistema. 2018-05-12 13: 04: 49.274137: W tensorflow / core / framework / allocator.cc: 101] La asignación de 19267584 excede el 10% de la memoria del sistema. 2018-05-12 13: 04: 49.647061: W tensorflow / core / framework / allocator.cc: 101] La asignación de 19267584 excede el 10% de la memoria del sistema. 2018-05-12 13: 04: 50.028839: W tensorflow / core / framework / allocator.cc: 101] La asignación de 19267584 excede el 10% de la memoria del sistema. 2018-05-12 13: 04: 50.413735: W tensorflow / core / framework / allocator.cc: 101] La asignación de 19267584 excede el 10% de la memoria del sistema.

¿Alguna idea para optimizar la forma de cargar el modelo pre-entrenado (o) deshacerse de este mensaje de advertencia?

¡Gracias