3c874575bf40e8b1fa2280371131a8f29ebb3e98,src/gluonnlp/models/albert.py,,,#,637

Before Change

    return cfg, tokenizer, local_params_path, local_mlm_params_path


BACKBONE_REGISTRY.register("albert", [AlbertModel,
                                      get_pretrained_albert,
                                      list_pretrained_albert])

After Change

    return cfg


PRETRAINED_URL = {
    "google_albert_base_v2": {
        "cfg": google_albert_base(),
        "spm_model": "google_albert_base_v2/spm-65999e5d.model",
        "vocab": "google_albert_base_v2/vocab-2ee53ae7.json",
        "params": "google_albert_base_v2/model-125be477.params",
        "mlm_params": "google_albert_base_v2/model_mlm-fe20650e.params",
        "lowercase": True,
    },
    "google_albert_large_v2": {
        "cfg": google_albert_large(),
        "spm_model": "google_albert_large_v2/spm-65999e5d.model",
        "vocab": "google_albert_large_v2/vocab-2ee53ae7.json",
        "params": "google_albert_large_v2/model-ad60bcd5.params",
        "mlm_params": "google_albert_large_v2/model_mlm-6a5015ee.params",
        "lowercase": True,
    },
    "google_albert_xlarge_v2": {
        "cfg": google_albert_xlarge(),
        "spm_model": "google_albert_xlarge_v2/spm-65999e5d.model",
        "vocab": "google_albert_xlarge_v2/vocab-2ee53ae7.json",
        "params": "google_albert_xlarge_v2/model-4149c9e2.params",
        "mlm_params": "google_albert_xlarge_v2/model_mlm-ee184d38.params",
        "lowercase": True,
    },
    "google_albert_xxlarge_v2": {
        "cfg": google_albert_xxlarge(),
        "spm_model": "google_albert_xxlarge_v2/spm-65999e5d.model",
        "vocab": "google_albert_xxlarge_v2/vocab-2ee53ae7.json",
        "params": "google_albert_xxlarge_v2/model-5601a0ed.params",
        "mlm_params": "google_albert_xxlarge_v2/model_mlm-d2e2b06f.params",
        "lowercase": True,
    },
}

FILE_STATS = load_checksum_stats(os.path.join(get_model_zoo_checksum_dir(), "albert.txt"))

Instances