Package smile.vision.layer
Class FusedMBConv
java.lang.Object
smile.deep.layer.LayerBlock
smile.vision.layer.FusedMBConv
Fused-MBConv replaces the depthwise-conv3×3 and expansion-conv1×1
in MBConv with single regular conv3×3.
-
Constructor Summary
ConstructorDescriptionFusedMBConv
(MBConvConfig config, double stochasticDepthProb, IntFunction<Layer> normLayer) Constructor. -
Method Summary
Methods inherited from class smile.deep.layer.LayerBlock
add, add, asTorch, toString
Methods inherited from class java.lang.Object
clone, equals, finalize, getClass, hashCode, notify, notifyAll, wait, wait, wait
Methods inherited from interface smile.deep.layer.Layer
apply, isTraining
-
Constructor Details
-
FusedMBConv
Constructor.- Parameters:
config
- block configuration.stochasticDepthProb
- the probability of the input to be zeroed in stochastic depth layer.normLayer
- the functor to create the normalization layer.
-
-
Method Details
-
forward
Description copied from interface:Layer
Forward propagation (or forward pass) through the layer.- Parameters:
input
- the input tensor.- Returns:
- the output tensor.
-