bucky527 / scme Goto Github PK

RuntimeError                              Traceback (most recent call last)
Cell In[15], line 7
      5 rnatorch,proteintorch=rnatorch.to(model.device),proteintorch.to(model.device)
      6 model.eval()
----> 7 zm=model.inference(rnatorch, proteintorch)

File ~/Workspace/Multiomics/benchmark/script/./scME/pyroMethod.py:653, in ScMESVI_2.inference(self, rna, protein)
    650 self.eval() 
    651 # zr_loc,zr_scale,l_loc,l_scale=self.zr_encoder((rna,yr))
    652 # zp_loc,zp_scale,c_loc,c_scale,pi=self.zp_encoder((protein,yp))
--> 653 zr_loc,zr_scale,l_loc,l_scale=self.zr_encoder(rna)
    654 zp_loc,zp_scale,c_loc,c_scale,pi=self.zp_encoder(protein)
    655 zm_loc,zm_scale=self.zm_encoder((zr_loc,zp_loc))

RuntimeError: mat1 and mat2 must have the same dtype

MLP(
  (sequential_mlp): Sequential(
    (0): ConcatModule()
    (1): DataParallel(
      (module): Linear(in_features=2000, out_features=1000, bias=True)
    )
    (2): BatchNorm1d(1000, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
    (3): ReLU()
    (4): DataParallel(
      (module): Linear(in_features=1000, out_features=256, bias=True)
    )
    (5): BatchNorm1d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
    (6): ReLU()
    (7): DataParallel(
      (module): Linear(in_features=256, out_features=64, bias=True)
    )
    (8): BatchNorm1d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
    (9): ReLU()
    (10): ListOutModule(
      (0): Sequential(
        (0): Linear(in_features=64, out_features=24, bias=True)
      )
      (1): Sequential(
        (0): Linear(in_features=64, out_features=24, bias=True)
        (1): Softplus(beta=1, threshold=20)
      )
      (2): Sequential(
        (0): Linear(in_features=64, out_features=1, bias=True)
      )
      (3): Sequential(
        (0): Linear(in_features=64, out_features=1, bias=True)
        (1): Softplus(beta=1, threshold=20)
      )
    )
  )
)