search

uw-ipd / RoseTTAFold2

MIT License
160 stars 36 forks source link

RuntimeError tensor size #33

Closed clemensau closed 5 months ago

clemensau commented 5 months ago

Hello,

Recently, a new error appeared :

File ".../RoseTTAFold2/network/predict.py", line 249, in predict
    xyz_t[:ntmpl_i,startres:stopres,:,:] = xyz_t_i
RuntimeError: The expanded size of the tensor (24) must 
match the existing size (2048) at non-singleton dimension 1. 
Target sizes: [4, 24, 27, 3].  Tensor sizes: [4, 2048, 27, 3]

Perhaps it's due to the new version of pytorch ? I've already added dim=-1 to torch.cross() in the network/util.py script.

Thanks in advance

clemensau commented 5 months ago

I applied what yamule said here https://github.com/uw-ipd/RoseTTAFold2/issues/21#issuecomment-1742029479 and it works !

Please note, https://github.com/uw-ipd/RoseTTAFold2/issues/21#issuecomment-1748988357