关于PyABSA v1手动加载本地checkpoint的技术咨询

针对PyABSA v1版本默认从Google Drive下载Checkpoint导致IP被封禁的问题，可通过以下步骤手动加载本地Checkpoint：

1. 获取对应版本的Checkpoint文件 #

找到与PyABSA v1兼容的目标模型Checkpoint（比如你使用的english ATEPC模型），将其下载到本地可访问的目录中。

2. 修改代码加载本地Checkpoint #

将原代码中在线加载Checkpoint的逻辑，替换为指定本地文件路径的方式。示例代码如下：

from pyabsa.functional import ATEPCCheckpointManager, ABSADatasetList

examples = ['But the staff was so nice to us .',
            'But the staff was so horrible to us .',
            r'Not only was the food outstanding , but the little ` perks \' were great .',
            'It took half an hour to get our check , which was perfect since we could sit , have drinks and talk !',
            'It was pleasantly uncrowded , the service was delightful , the garden adorable , '
            'the food -LRB- from appetizers to entrees -RRB- was delectable .',
            'How pretentious and inappropriate for MJ Grill to claim that it provides power lunch and dinners !'
            ]

inference_source = ABSADatasetList.Restaurant14
# 将路径替换为你的本地Checkpoint文件路径，例如'./atepc_english_v1_checkpoint.pth'
aspect_extractor = ATEPCCheckpointManager.get_aspect_extractor(checkpoint='/your/local/checkpoint/path')
atepc_result = aspect_extractor.extract_aspect(inference_source=inference_source,
                                               save_result=True,
                                               print_result=True,
                                               pred_sentiment=True,
                                               )

注意事项 #

• 确保下载的Checkpoint版本与PyABSA v1完全兼容，避免版本不匹配导致加载失败。
• 本地路径需使用正确的绝对路径或相对路径，防止代码无法定位文件。

内容的提问来源于stack exchange，提问作者Rose_Trojan