VGGBase

Swift

associatedtype Conv1 : LayerType where Self.Device == Self.Conv1.Device, Self.Parameter == Self.Conv1.Parameter, Self.Conv1.Device == Self.Conv2.Device, Self.Conv1.Inputs == Tensor<Self.Parameter, Self.Device>, Self.Conv1.Outputs == Tensor<Self.Parameter, Self.Device>, Self.Conv1.Parameter == Self.Conv2.Parameter

Swift

associatedtype Conv2 : LayerType where Self.Conv2.Device == Self.Conv3.Device, Self.Conv2.Inputs == Tensor<Self.Parameter, Self.Device>, Self.Conv2.Outputs == Tensor<Self.Parameter, Self.Device>, Self.Conv2.Parameter == Self.Conv3.Parameter

Swift

associatedtype Conv3 : LayerType where Self.Conv3.Device == Self.Conv4.Device, Self.Conv3.Inputs == Tensor<Self.Parameter, Self.Device>, Self.Conv3.Outputs == Tensor<Self.Parameter, Self.Device>, Self.Conv3.Parameter == Self.Conv4.Parameter

Swift

associatedtype Conv4 : LayerType where Self.Conv4.Device == Self.Conv5.Device, Self.Conv4.Inputs == Tensor<Self.Parameter, Self.Device>, Self.Conv4.Outputs == Tensor<Self.Parameter, Self.Device>, Self.Conv4.Parameter == Self.Conv5.Parameter

Swift

associatedtype Conv5 : LayerType where Self.Conv5.Inputs == Tensor<Self.Parameter, Self.Device>, Self.Conv5.Outputs == Tensor<Self.Parameter, Self.Device>

Swift

typealias DenseLayer = Sequential<Sequential<Sequential<Sequential<Dense<Self.Parameter, Self.Device>, BatchNorm<Self.Parameter, Self.Device>>, Sequential<Relu<Self.Parameter, Self.Device>, Dropout<Self.Parameter, Self.Device>>>, Sequential<Sequential<Dense<Self.Parameter, Self.Device>, BatchNorm<Self.Parameter, Self.Device>>, Sequential<Relu<Self.Parameter, Self.Device>, Dropout<Self.Parameter, Self.Device>>>>, Sequential<Dense<Self.Parameter, Self.Device>, LogSoftmax<Self.Parameter, Self.Device>>>

Swift

var conv1: Conv1 { get set }

Swift

var conv2: Conv2 { get set }

Swift

var conv3: Conv3 { get set }

Swift

var conv4: Conv4 { get set }

Swift

var conv5: Conv5 { get set }

Swift

var dense: DenseLayer { get set }

Swift

var parameters: [Tensor<Parameter, Self.Device>] { get }

Swift

var parameterPaths: [WritableKeyPath<Self, Tensor<Parameter, Device>>] { get }

Swift

var isDropoutActive: Bool { get set }

Swift

func callAsFunction(_ inputs: Tensor<Parameter, Device>) -> Tensor<Parameter, Device>

Swift

static func makeDense(classes: Int) -> DenseLayer