まず、Github のプロンプトに従って環境を作成します
cd CronKGQA
conda create --prefix ./cronkgqa_env python=3.8
conda activate ./cronkgqa_env
バージョンの不一致 1.7.1 の pytorch を再インストールします。
conda install pytorch==1.7.1 torchvision==0.8.2 cudatoolkit=11.0 -c pytorch
cronkgqa_testing.ipynb
テキストによる質問に対するモデルの応答をテストするために使用できます。(エンティティと時間のリストも質問に記載されています)
1. RuntimeError: モデルの state_dict のロード中にエラーが発生しました: state_dict にキーがありません
改訂:
qa_model.load_state_dict(torch.load(filename), False)
load_state_dict メソッドのパラメータの公式説明 strict パラメータのデフォルトは true で、これは、state_dict 内のキーがモジュールのキーによって返されるキーと一致することを厳密に要求するかどうかを意味します。strict が false に設定されている場合、エラー レポートは無視できますが、無視されることに注意してください。
参考:ブログ
2. 出力がサイズ制限を超えています。完全な出力データをテキスト エディタで開きます。
ノート
警告モジュールを使用して警告を無視する
import warnings
warnings.filterwarnings("ignore")#程序中
python -W ignore xx.py # 命令行