🐛 Describe the bug I try to train a GNN network based on Transformer_Conv. It worked well on GPU RTX 3090 but failed on GPU RTX 4090. I also tested other GNN Convs like GCNConv or GATConv, GCNConv performed well but GATConv not. It seems...
model = SentenceTransformer(model_name, device=device) # 嵌入模型不参与后续图神经网络的训练 @torch.no_grad() def __call__(self, df): x = self.model.encode( # 要进行嵌入的值 df.values, # 显示处理进度 show_progress_bar=True, # 转换为PyTorch的张量 convert_to_tensor=True, # 使用的...