yilye

CDFA_model.py insert 93 line: self.dist = torch.transpose(self.dist, 1, 3) ... self.dist = F.pairwise_distance(self.feat_A, self.feat_B, keepdim=True) # 特征距离 self.dist = torch.transpose(self.dist, 1, 3) self.dist = F.interpolate(self.dist, size=self.A.shape[2:], mode='bilinear',align_corners=True) ...

1.python preprocess.py -train_src xxx -train_tgt xxx -dev_src xxx -dev_tgt xxx -save_data 222 2.python train.py -data_path 222 -model_path 333 3.w_q = Linear([d_model, d_k * n_heads]) ----> w_q = Linear(d_model, d_k *...