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 a GhostBatchNorm.

Field Summary

Fields inherited from class ai.djl.nn.norm.BatchNorm.BaseBuilder

axis, center, epsilon, momentum, scale

Method Summary

Modifier and Type	Method	Description
GhostBatchNorm	build()	Builds the new GhostBatchNorm.
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

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

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 new GhostBatchNorm.

Specified by:

build in class BatchNorm.BaseBuilder<GhostBatchNorm.Builder>

Returns:

the new GhostBatchNorm

self

public GhostBatchNorm.Builder self()

Returns this {code Builder} object.

Specified by:

self in class BatchNorm.BaseBuilder<GhostBatchNorm.Builder>

Returns:

this BaseBuilder