Open zyy1234q opened 4 months ago
您好作者在你的SC2_PCR.py代码中第336行 //distance = torch.sqrt(2 - 2 (src_desc[0] @ tgt_desc[0].T) + 1e-6)这里不应该是distance = torch.sqrt(2 - 2 (src_desc @ tgt_desc.T) + 1e-6)吗,为什么要用第一个批次形成的粗匹配去做
您好作者在你的SC2_PCR.py代码中第336行 //distance = torch.sqrt(2 - 2 (src_desc[0] @ tgt_desc[0].T) + 1e-6)这里不应该是distance = torch.sqrt(2 - 2 (src_desc @ tgt_desc.T) + 1e-6)吗,为什么要用第一个批次形成的粗匹配去做 不好意思我还以为训练呢,这是test batch_size=1
您好作者在你的SC2_PCR.py代码中第336行 //distance = torch.sqrt(2 - 2 (src_desc[0] @ tgt_desc[0].T) + 1e-6)这里不应该是distance = torch.sqrt(2 - 2 (src_desc @ tgt_desc.T) + 1e-6)吗,为什么要用第一个批次形成的粗匹配去做