strawberryfg / GroupNorm-caffe

caffe implementation of Group Normalization https://arxiv.org/abs/1803.08494

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

GroupNorm-caffe

caffe implementation of Group Normalization https://arxiv.org/abs/1803.08494

This code only provide CUDA version

add to caffe.proto

message GroupNormParameter {

optional float eps = 1 [default = 1e-5];

optional int32 group_num = 2 [default = 32];

}

optional GroupNormParameter group_norm_param = 10000;

About

caffe implementation of Group Normalization https://arxiv.org/abs/1803.08494


Languages

Language:Cuda 60.6%Language:C++ 39.4%