uw-ipd / RoseTTAFold2

MIT License
170 stars 39 forks source link

AttributeError: 'ConvSE3' object has no attribute 'weights' #9

Closed aravinda1879 closed 1 year ago

aravinda1879 commented 1 year ago

Hi, I tried running the first example and got the "'ConvSE3' object has no attribute 'weights'" error. I did use the SE3Transformer in this repo. Any directions are highly appreciated. Thanks!

Command: ../run_RF2.sh rcsb_pdb_7UGF.fasta -o 7UGF

error: Running RoseTTAFold2 to predict structures -> Running command: python RoseTTAFold2/network/predict.py -inputs 7UGF/rcsb_pdb_7UGF_1.msa0.a3m -prefix 7UGF/models/model -model /software/RoseTTAFold2/network/weights/RF2_apr23.pt -db /software/RoseTTAFold2/pdb100_2021Mar03/pdb100_2021Mar03 -symm C1 /software/RoseTTAFold2/network/weights/RF2_apr23.pt Running on GPU Traceback (most recent call last): File "/software/RoseTTAFold2/network/predict.py", line 488, in pred = Predictor(args.model, torch.device("cuda:0")) File "/software/RoseTTAFold2/network/predict.py", line 165, in init self.model = RoseTTAFoldModule( File "/software/RoseTTAFold2/network/RoseTTAFoldModel.py", line 30, in init self.simulator = IterativeSimulator(n_extra_block=n_extra_block, File "/software/RoseTTAFold2/network/Track_module.py", line 591, in init self.extra_block = nn.ModuleList([IterBlock(d_msa=d_msa_full, d_pair=d_pair, File "/software/RoseTTAFold2/network/Track_module.py", line 591, in self.extra_block = nn.ModuleList([IterBlock(d_msa=d_msa_full, d_pair=d_pair, File "/software/RoseTTAFold2/network/Track_module.py", line 541, in init self.str2str = Str2Str(d_msa=d_msa, d_pair=d_pair, File "/software/RoseTTAFold2/network/Track_module.py", line 449, in init self.se3 = SE3TransformerWrapper(**SE3_param) File "/software/RoseTTAFold2/network/SE3_network.py", line 58, in init self.reset_parameter() File "/software/RoseTTAFold2/network/SE3_network.py", line 82, in resetparameter nn.init.zeros(self.se3.graph_modules[-1].weights['0']) File "/software/RFDesign/envs/.conda/envs/rfdesign/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1130, in getattr raise AttributeError("'{}' object has no attribute '{}'".format( AttributeError: 'ConvSE3' object has no attribute 'weights'