Open drcege opened 1 year ago
After skimming through the code, I cannot find the relevant part of gradient projection as Eq. 1 in the paper.
If it's my carelessness, could you please help me figure it out?
I think its here: https://github.com/facebookresearch/NASViT/blob/0472916056803ddb5c8f9c2d93ef943e0c99d026/misc/constrain_opt.py#L8
After skimming through the code, I cannot find the relevant part of gradient projection as Eq. 1 in the paper.
If it's my carelessness, could you please help me figure it out?