update useage
This commit is contained in:
15
README.md
15
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
|
||||
```
|
||||
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
|
||||
```
|
||||
|
||||
Reference in New Issue
Block a user