diff --git a/configs/__init__.py b/configs/__init__.py index 6f9aec3..71787ae 100644 --- a/configs/__init__.py +++ b/configs/__init__.py @@ -1,7 +1,7 @@ pretrained_models_path = { - 'bert-base-uncased': '/home/sharing/disk1/disk1/pretrained_embedding/bert/bert-base-uncased', - 'bert-large-uncased':'/home/sharing/disk1/disk1/pretrained_embedding/bert/bert-large-uncased', + 'bert-base-uncased': 'google-bert/bert-base-uncased', + 'bert-large-uncased':'google-bert/bert-large-uncased', } feat_dims = { @@ -9,4 +9,4 @@ 'bert-base-uncased': 768, 'bert-large-uncased': 1024 } -} \ No newline at end of file +}