Commit Graph

3 Commits (ee634df2900996233473cb95a80029bd456cce97)

Author SHA1 Message Date
liaoxingyu 3840f3f79a fix arcface NaN problem
Summary: fix classifier init bugs, which will not initialize classifier weights when use arcface or circle loss.
In this way, it will lead loss NaN problem.
2020-06-18 12:05:44 +08:00
liaoxingyu 5528d17ace refactor code
Summary: change code style and refactor code, add avgmax pooling layer in gem_pool
2020-05-28 13:49:39 +08:00
liaoxingyu a2dcd7b4ab feat(layers/norm): add ghost batchnorm
add a get_norm fucntion to easily change normalization between batchnorm, ghost bn and group bn
2020-05-01 09:02:46 +08:00