Package ai.djl.nn.norm
Class GhostBatchNorm.Builder
- java.lang.Object
-
- ai.djl.nn.norm.BatchNorm.BaseBuilder<GhostBatchNorm.Builder>
-
- ai.djl.nn.norm.GhostBatchNorm.Builder
-
- Enclosing class:
- GhostBatchNorm
public static class GhostBatchNorm.Builder extends BatchNorm.BaseBuilder<GhostBatchNorm.Builder>
The Builder to construct aGhostBatchNorm
.
-
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description GhostBatchNorm
build()
Builds the newGhostBatchNorm
.GhostBatchNorm.Builder
optVirtualBatchSize(int virtualBatchSize)
Sets the size of virtual batches in which to use when sub-batching.GhostBatchNorm.Builder
self()
Returns this {code Builder} object.-
Methods inherited from class ai.djl.nn.norm.BatchNorm.BaseBuilder
optAxis, optCenter, optEpsilon, optMomentum, optScale
-
-
-
-
Method Detail
-
optVirtualBatchSize
public GhostBatchNorm.Builder optVirtualBatchSize(int virtualBatchSize)
Sets the size of virtual batches in which to use when sub-batching. Defaults to 128.- Parameters:
virtualBatchSize
- the virtual batch size- Returns:
- this Builder
-
build
public GhostBatchNorm build()
Builds the newGhostBatchNorm
.- Specified by:
build
in classBatchNorm.BaseBuilder<GhostBatchNorm.Builder>
- Returns:
- the new
GhostBatchNorm
-
self
public GhostBatchNorm.Builder self()
Returns this {code Builder} object.- Specified by:
self
in classBatchNorm.BaseBuilder<GhostBatchNorm.Builder>
- Returns:
- this
BaseBuilder
-
-