Carga de pesos en formato TH cuando keras está configurado en formato TF
Tengo Kerasimage_dim_ordering
propiedad establecida en 'tf', por lo que defino mis modelos de esta manera:
model = Sequential()
model.add(ZeroPadding2D((1, 1), input_shape=(224, 224, 3)))
model.add(Convolution2D(64, 3, 3, activation='relu'))
Pero cuando llamoload_weights
método, se bloquea porque mi modelo se guardó con el formato "th":
Exception: Layer weight shape (3, 3, 3, 64) not compatible with provided weight shape (64, 3, 3, 3)
¿Cómo puedo cargar estos pesos y transponerlos automáticamente para corregir el formato de Tensorflow?