FaceX-Zoo
FaceX-Zoo copied to clipboard
A problem I met when i use the 'Mv-softmax' as head:one of the variables needed for gradient computation has been modified by an inplace operation: [torch.cuda.FloatTensor
when i use the 'Mv-softmax' as head to train a model,i met the problem:'one of the variables needed for gradient computation has been modified by an inplace operation: [torch.cuda.FloatTensor [64, 10575]], which is output 0 of MmBackward, is at version 3; expected version 0 instead. Hint: enable anomaly detection to find the operation that failed to compute its gradient, with torch.autograd.set_detect_anomaly(True).'how can i solve the problem? thank you