search

facebookresearch / NASViT

code for NASViT
Other
66 stars 3 forks source link

Implementation of gradient projection #8

Open drcege opened 1 year ago

drcege commented 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?

uzair789 commented 1 year ago

I think its here: https://github.com/facebookresearch/NASViT/blob/0472916056803ddb5c8f9c2d93ef943e0c99d026/misc/constrain_opt.py#L8