RuntimeError: mat1 and mat2 shapes cannot be multiplied (123x18 and 19x56)

luwei0917 / TankBind

Open source code for TankBind. Galixir Tenchnologies

MIT License

150 stars 49 forks source link

RuntimeError: mat1 and mat2 shapes cannot be multiplied (123x18 and 19x56) #10

Closed zhengyangkust closed 2 years ago

zhengyangkust commented 2 years ago

Hi Wei,

I ran the example prediction code and get error when using the 'affinity_pred_list = torch.cat(affinity_pred_list)', I don't know what caused error and how to fix it.

Thanks!

1657422948598

luwei0917 commented 2 years ago

please use torchdrug v0.1.2