mirror of
https://github.com/opencv/opencv.git
synced 2025-06-13 04:52:53 +08:00
support silu activation in darknet
This commit is contained in:
parent
925ff6241f
commit
5e92bf8e41
@ -229,7 +229,7 @@ namespace cv {
|
|||||||
activation_param.set<float>("negative_slope", 0.1f);
|
activation_param.set<float>("negative_slope", 0.1f);
|
||||||
activation_param.type = "ReLU";
|
activation_param.type = "ReLU";
|
||||||
}
|
}
|
||||||
else if (type == "swish")
|
else if (type == "swish" || type == "silu") // swish is an extension of silu.
|
||||||
{
|
{
|
||||||
activation_param.type = "Swish";
|
activation_param.type = "Swish";
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user