ai.djl.basicmodelzoo.cv.classification

Class ResnetBlockFactory

java.lang.Object

ai.djl.basicmodelzoo.cv.classification.ResnetBlockFactory

All Implemented Interfaces:

ai.djl.nn.BlockFactory, java.io.Serializable

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

A BlockFactory class that creates ResNetV1 block.

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
ResnetBlockFactory()

Method Summary

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

Methods inherited from class java.lang.Object

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

Constructor Detail

ResnetBlockFactory

public ResnetBlockFactory()

Method Detail

newBlock

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

Specified by:

newBlock in interface ai.djl.nn.BlockFactory