search

ShenAoAO / PGBind

3 stars 1 forks source link

Question for FABind_PG #1

Open lyy-369 opened 1 week ago

lyy-369 commented 1 week ago

Hello, when I use FABind_PG and perform 'Evaluation', there is a problem using model.safetensors in the loaded model. Which model is this? Is there a more accurate version for learning?

ShenAoAO commented 1 week ago

Did you install safetensors?

lyy-369 commented 1 week ago

Yes, I commented line 234 in 'FABind_PG/fabind/test_fabind.py' and read model.safetensors using line 233. I found that the model did not match. How should I modify it? Traceback (most recent call last): File "/home/lyy/Desktop/PGBind/FABind_PG/fabind/test_fabind.py", line 233, in load_model(model,'../ckpt/model.safetensors') File "/home/lyy/anaconda3/envs/FABind/lib/python3.8/site-packages/safetensors/torch.py", line 221, in load_model raise RuntimeError(error) RuntimeError: Error(s) in loading state_dict for IaBNet_mean_and_pocket_prediction_cls_coords_dependent: Unexpected key(s) in state_dict: "transformer.embedding.embedding.div_term", "transformer.embedding.proj_a.bias", "transformer.embedding.proj_a.weight", ...

ShenAoAO commented 6 days ago

import model_original改成model_n,我做消融的时候忘记改回来了