AlexNet.Builder (Deep Java Library 0.21.0 - model-zoo API)

java.lang.Object
- ai.djl.basicmodelzoo.cv.classification.AlexNet.Builder

Enclosing class:: AlexNet

public static final class AlexNet.Builder
extends java.lang.Object

The Builder to construct a AlexNet object.

Method Summary

All Methods Instance Methods Concrete Methods
Modifier and Type	Method	Description
`ai.djl.nn.Block`	`build()`	Builds a `AlexNet` block.
`AlexNet.Builder`	`setDropOutRate(float dropOutRate)`	Sets the dropout rate in the network.
`AlexNet.Builder`	`setNumChannels(int[] numChannels)`	Sets the number of channels for the AlexNet blocks.
`AlexNet.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 Detail
  - setDropOutRate
```
public AlexNet.Builder setDropOutRate(float dropOutRate)
```
    Sets the dropout rate in the network.
    
    Parameters:
    
    dropOutRate - the dropout rate
    
    Returns:
    
    this Builder
  - setNumChannels
```
public AlexNet.Builder setNumChannels(int[] numChannels)
```
    Sets the number of channels for the AlexNet blocks.
    
    Parameters:
    
    numChannels - the number of channels for every AlexNet block.
    
    Returns:
    
    this Builder
  - setOutSize
```
public AlexNet.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 AlexNet block.
    
    Returns:
    
    the AlexNet block