尝试重命名tf.keras上的预训练模型时出错

import tensorflow as tf

with tf.variable_scope('my_new_scope'):
    base_model = tf.keras.applications.MobileNetV2(input_shape=(224, 224, 3), include_top=False)
    # 现在模型的某些部分会带有'my_new_scope'前缀

import tensorflow as tf

class CustomModel(tf.keras.Model):
    def __init__(self, base_model):
        super(CustomModel, self).__init__()
        self.base_model = base_model
        # 重命名模型的层
        for layer in self.base_model.layers:
            layer._name = layer.name.replace('base_model_', 'custom_model_')

base_model = tf.keras.applications.MobileNetV2(input_shape=(224, 224, 3), include_top=False)
custom_model = CustomModel(base_model)

import tensorflow as tf

base_model = tf.keras.applications.MobileNetV2(input_shape=(224, 224, 3), include_top=False)
base_model.save('temp_model')

loaded_model = tf.keras.models.load_model('temp_model', compile=False)
loaded_model._name = 'custom_model_name'