diff --git a/README.md b/README.md index 7fb2726..4ba86d5 100644 --- a/README.md +++ b/README.md @@ -7,5 +7,16 @@ micromamba create -n admet_ai python=3.10 micromamba activate admet_ai pip install -r requirements.txt pip install -e . -python fix_admet_load.py --data_path ./data/molecules.csv --save_path ./data/preds.csv --smiles_column smiles -``` \ No newline at end of file +admet_predict --data_path ./data/molecules.csv --save_path ./data/preds.csv --smiles_column smiles +``` + +## pytorch 降级 + +看起来你遇到的是 PyTorch 2.6+ 把 torch.load 的默认参数改成了 weights_only=True 导致的非兼容问题。admet-ai(通过 chemprop 的 load_args)在加载 checkpoint 里保存的 argparse.Namespace 时没有显式设 weights_only=False,因此触发了 PyTorch 的安全反序列化报错。详见官方说明:2.6 起默认改动与报错指引。 + +```bash +micromamba activate admet_ai +python -c "import torch,sys;print(torch.__version__)" +pip install --upgrade 'torch<2.6' +pip install --upgrade 'torch<2.6' --index-url https://download.pytorch.org/whl/cu121 +```