Package ai.djl.basicmodelzoo.cv.classification

Class ResnetBlockFactory

java.lang.Object

ai.djl.basicmodelzoo.cv.classification.ResnetBlockFactory

All Implemented Interfaces:

ai.djl.nn.BlockFactory, Serializable

public class ResnetBlockFactory
extends Object
implements ai.djl.nn.BlockFactory

A BlockFactory class that creates ResNetV1 block.

See Also:

• Serialized Form

Constructor Summary

Constructors

Constructor	Description
ResnetBlockFactory()

Method Summary

Modifier and Type	Method	Description
ai.djl.nn.Block	newBlock(ai.djl.Model model, Path modelPath, Map<String,?> arguments)

Methods inherited from class java.lang.Object

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

Constructor Details

ResnetBlockFactory

public ResnetBlockFactory()

Method Details

newBlock

public ai.djl.nn.Block newBlock(ai.djl.Model model,
 Path modelPath,
 Map<String,?> arguments)

Specified by:

newBlock in interface ai.djl.nn.BlockFactory