Open xysong1201 opened 4 years ago
I meet the same problem when I train the code on multi server, but I don't know the reason
I solve the problem, Multi Sever can not use mask operation, you can use torch.where to replace it
cos_theta_hard = (self.t + 1.0) * cos_theta + self.t
cos_theta = torch.where(cos_theta > gt -self.margin, cos_theta_hard, cos_theta)
Hi, I tried your MV-AM and MV-Arc , both give me this error. Someone encountered the same error?