kharrigian / ehr-stigma Goto Github PK

Thank you so much for the great work!

For anyone who encountered the following error when initializing StigmaBertModel:
Traceback (most recent call last):
File "C:\Users\Louis\miniconda3\envs\SL-infer\lib\code.py", line 90, in runcode
exec(code, self.locals)
File "", line 2, in
File "C:\Users\Louis\PycharmProjects\pythonProject5\ehr-stigma-main\stigma\api.py", line 381, in init
_ = self._initialize_model(self._model)
File "C:\Users\Louis\PycharmProjects\pythonProject5\ehr-stigma-main\stigma\api.py", line 485, in _initialize_model
_ = self.model_dict["classifier"].load_state_dict(torch.load(f"{model}/model.pth", map_location=torch.device('cpu')))
File "C:\Users\Louis\miniconda3\envs\SL-infer\lib\site-packages\torch\nn\modules\module.py", line 2152, in load_state_dict
raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format(
RuntimeError: Error(s) in loading state_dict for BERTMultitaskClassifier:
Unexpected key(s) in state_dict: "_bert._bert.embeddings.position_ids".

A quick fix that works for me is to add strict=False at line #485 of the stigma/api.py, i.e.:
_ = self.model_dict["classifier"].load_state_dict(torch.load(f"{model}/model.pth", map_location=torch.device('cpu')), strict=False)

I hope this helps! Thanks!