diff --git a/psenet/layers.h b/psenet/layers.h index bd85a18..264d645 100644 --- a/psenet/layers.h +++ b/psenet/layers.h @@ -13,6 +13,6 @@ IScaleLayer *addBatchNorm2d(INetworkDefinition *network, std::map &weightMap, ITensor &input, int ch, int stride, std::string lname, int branch_type); -IActivationLayer *ConvRelu(INetworkDefinition *network, std::map &weightMap, ITensor &input, int outch, int kernel, int stride, std::string lname); +IActivationLayer *addConvRelu(INetworkDefinition *network, std::map &weightMap, ITensor &input, int outch, int kernel, int stride, std::string lname); #endif