docs
0.0-2dfa388-SNAPSHOT
API
Generated with
docs/torchvision/torchvision.models/resnet/ResNet

ResNet

torchvision.models.resnet.ResNet
class ResNet[D <: BFloat16 | Float32 | Float64](block: BlockBuilder, layers: Seq[Int], numClasses: Int, zeroInitResidual: Boolean, groups: Int, widthPerGroup: Int, replaceStrideWithDilation: (Boolean, Boolean, Boolean))(using x$8: Default[D])(normLayer: Int => HasWeight[D] & TensorModule[D]) extends Module

Attributes

Source
resnet.scala
Graph
Supertypes
class Module
class Object
trait Matchable
class Any

Members list

Value members

Concrete methods

def apply(x: Tensor[D]): Tensor[D]

Attributes

Source
resnet.scala

Inherited methods

def apply(fn: Module => Unit): Module.this.type

Attributes

Inherited from:
Module
Source
Module.scala
def eval(): Unit

Attributes

Inherited from:
Module
Source
Module.scala
def hasBias(): Boolean

Attributes

Inherited from:
Module
Source
Module.scala
def isTraining: Boolean

Attributes

Inherited from:
Module
Source
Module.scala
def load(inputArchive: InputArchive): Unit

Attributes

Inherited from:
Module
Source
Module.scala
def loadStateDict(stateDict: Map[String, Tensor[DType]]): Unit

Attributes

Inherited from:
Module
Source
Module.scala
def modules: Seq[Module]

Attributes

Inherited from:
Module
Source
Module.scala
def modules(recurse: Boolean): Seq[Module]

Attributes

Inherited from:
Module
Source
Module.scala
def namedBuffers(recurse: Boolean): SeqMap[String, Tensor[_]]

Attributes

Inherited from:
Module
Source
Module.scala
def namedChildren: SeqMap[String, Module]

Attributes

Inherited from:
Module
Source
Module.scala
def namedModules: SeqMap[String, Module]

Attributes

Inherited from:
Module
Source
Module.scala
def namedParameters(recurse: Boolean): SeqMap[String, Tensor[_]]

Attributes

Inherited from:
Module
Source
Module.scala
def parameters(recurse: Boolean): Seq[Tensor[_]]

Attributes

Inherited from:
Module
Source
Module.scala
def parameters: Seq[Tensor[_]]

Attributes

Inherited from:
Module
Source
Module.scala
def register[M <: Module](child: M, n: String)(using name: Name): M

Attributes

Inherited from:
Module
Source
Module.scala
def registerBuffer[D <: DType](name: String, tensor: Tensor[D]): Tensor[D]

Adds a buffer to the module.

Adds a buffer to the module.

Attributes

Inherited from:
Module
Source
Module.scala
def registerBuffer[D <: DType](t: Tensor[D], n: String)(using name: Name): Tensor[D]

Attributes

Inherited from:
Module
Source
Module.scala
def registerModule[M <: Module](child: M, n: String)(using name: Name): M

Attributes

Inherited from:
Module
Source
Module.scala
def registerParameter[D <: DType](t: Tensor[D], requiresGrad: Boolean, n: String)(using name: Name): Tensor[D]

Attributes

Inherited from:
Module
Source
Module.scala
def save(outputArchive: OutputArchive): Unit

Attributes

Inherited from:
Module
Source
Module.scala
def summarize: String

Attributes

Inherited from:
Module
Source
Module.scala
def to(device: Device): Module.this.type

Attributes

Inherited from:
Module
Source
Module.scala
override def toString(): String

Returns a string representation of the object.

Returns a string representation of the object.

The default representation is platform dependent.

Attributes

Returns

a string representation of the object.

Definition Classes
Module -> Any
Inherited from:
Module
Source
Module.scala
def train(on: Boolean): Unit

Attributes

Inherited from:
Module
Source
Module.scala

Concrete fields

val avgpool: AdaptiveAvgPool2d[D]

Attributes

Source
resnet.scala
val baseWidth: Int

Attributes

Source
resnet.scala
val bn1: HasWeight[D] & TensorModule[D]

Attributes

Source
resnet.scala
val conv1: Conv2d[D]

Attributes

Source
resnet.scala
var dilation: Int

Attributes

Source
resnet.scala
val fc: Linear[D]

Attributes

Source
resnet.scala
var inplanes: Int

Attributes

Source
resnet.scala
val layer1: Sequential[D]

Attributes

Source
resnet.scala
val layer2: Sequential[D]

Attributes

Source
resnet.scala
val layer3: Sequential[D]

Attributes

Source
resnet.scala
val layer4: Sequential[D]

Attributes

Source
resnet.scala
val maxpool: MaxPool2d[D]

Attributes

Source
resnet.scala
val relu: ReLU[D]

Attributes

Source
resnet.scala
Generated with