Я пытался воспроизвести математическую модель рассуждения с помощью RAG.
Я хочу запустить этап прогнозирования ее модели поиска. Первоначально она обучалась с использованием Google byt5-small. p>
Итак, сначала я пишу скрипт py для переноса файла исходной модели из .safetensors в .ckpt.
затем я настраиваю переданный файл .ckpt в качестве контрольной точки (поскольку модель это модуль Pytorch Lightning)
тогда написано:
self.trainer.strategy.load_model_state_dict(
15944 File "/root/miniconda3/envs/lean/lib/python3.11/site-packages/pytorch_lightning/strategies/strategy.py", line 371, in load_model_state_dict
15945 self.lightning_module.load_state_dict(checkpoint["state_dict"], strict=strict)
15946 File "/root/miniconda3/envs/lean/lib/python3.11/site-packages/torch/nn/modules/module.py", line 2215, in load_state_dict
15947 raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format(
15948 RuntimeError: Error(s) in loading state_dict for PremiseRetriever:
15949 Missing key(s) in state_dict:....
...
Unexpected key(s) in state_dict: ....
Подробнее здесь: https://stackoverflow.com/questions/789 ... state-dict
