def variable_mapping(self):
    mapping = super(ELECTRA, self).variable_mapping()
    mapping['Embedding-Mapping'] = ['electra/embeddings_project/kernel', 'electra/embeddings_project/bias']
    mapping = {k: [i.replace('bert/', 'electra/') for i in v] for (k, v) in mapping.items()}
    mapping['Discriminator-Dense'] = ['discriminator_predictions/dense/kernel', 'discriminator_predictions/dense/bias']
    mapping['Discriminator-Prediction'] = ['discriminator_predictions/dense_1/kernel', 'discriminator_predictions/dense_1/bias']
    return mapping
def variable_mapping(self):
    mapping = super(ELECTRA, self).variable_mapping()
    mapping['Embedding-Mapping'] = ['electra/embeddings_project/kernel', 'electra/embeddings_project/bias']
    mapping = {k: [i.replace('bert/', 'electra/') for i in v] for (k, v) in mapping.items()}
    mapping['Discriminator-Dense'] = ['discriminator_predictions/dense/kernel', 'discriminator_predictions/dense/bias']
    mapping['Discriminator-Prediction'] = ['discriminator_predictions/dense_1/kernel', 'discriminator_predictions/dense_1/bias']
    return mapping