Проблема с сохранением модели Tensorflow2

Question

Сейчас я использую метрику Arcface для магистрали Res50, и у меня возникли проблемы с сохранением файла модели.

Отображаемая ошибка:

Слои с аргументами в init должны переопределять get_config.

полное сообщение об ошибке ниже. введите описание изображения здесь

что может потребовать переопределения конфигурации модели, и я не совсем уверен в этом.

Я сослался на функцию vgg_arcface из репозитория https://github.com/4uiiurz1/keras-arcface, чтобы сделать функцию Res50_arcface, а под ней результат.

# Resnet Backbone
def ResNet50_arcface(args):
    y = Input(shape=(200,))
    x = inputs = Input([224, 224, 3], name='input_image')
    x = ResNet50(input_shape=x.shape[1:], include_top=False, weights='imagenet')(x)

    # BN + Dropout + FC + BN ( as paper mentioned )
    x = BatchNormalization()(x)
    x = Dropout(rate=0.5)(x)
    x = Flatten()(x)
    x = Dense(args.num_features, kernel_regularizer=regularizers.l2(5e-4))(x)
    x = BatchNormalization()(x)

    output = ArcFace(200, regularizer=regularizers.l2(weight_decay))([x, y])

    return Model([inputs, y], output)

Может ли кто-нибудь дать мне предложение?

Спасибо