Package ai.djl.basicmodelzoo.cv.classification

Class GoogLeNet.Builder

java.lang.Object

ai.djl.basicmodelzoo.cv.classification.GoogLeNet.Builder

Enclosing class:

GoogLeNet

public static final class GoogLeNet.Builder
extends Object

The Builder to construct a GoogLeNet object.

Method Summary

Modifier and Type	Method	Description
ai.djl.nn.Block	build()	Builds a GoogLeNet block.
GoogLeNet.Builder	setOutSize(long outSize)	Sets the size of the output.

Methods inherited from class java.lang.Object

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

Method Details

setOutSize

public GoogLeNet.Builder setOutSize(long outSize)

Sets the size of the output.

Parameters:

outSize - the output size

Returns:

this Builder

build

public ai.djl.nn.Block build()

Builds a GoogLeNet block.

Returns:

the GoogLeNet block