search

zma-c-137 / VarGFaceNet

312 stars 84 forks source link

missing a se block in head setting #6

Closed TatsuyaShirakawa closed 5 years ago

TatsuyaShirakawa commented 5 years ago

It seems to have a se block at the end of branch in head setting (i.e. varg-conv1x1-varg-conv1x1-se) but it's not visualized in the network diagram.

image

More detailed network diagram is attached VargFaceNet.pdf which is created by the following code. 

    net = get_symbol()
    digraph = mx.viz.plot_network(net, shape={'data': (1, 3, 112, 112)},
                                  node_attrs={'fixedsize': 'false'})
    digraph.view()

Thanks.

zma-c-137 commented 5 years ago

yes, there is a se block at the end of head setting. Thank you for pointing out this mistake and we will update our arxiv paper as well as this repo soon.