JaminFong/DenseNAS

Shape alignement code

theevann opened this issue · 1 comments

Hi !
I am interested in the shape-alignement layer and I would like to know if you could share a snippet of code for this layer ? This part is unfortunately not detailed in the paper.
Thank you much !

@theevann Hi! Thanks for your attention to our work. The code of the search space is released. The shape-alignment layer is defined as HeadLayer in the code for simplicity. Hoping this can help you.

class HeadLayer(nn.Module):