模型文件第95行代码 feat = torch.cat([feat, gf_feat.repeat(1, feat.size(2), 1)], dim=-1) # B,640,512中的feat.size(2)是不是应该写成feat.size(1)
