VGG11

Swift

public typealias Parameter = E

Swift

public typealias Device = D

Swift

public var conv1: Sequential<Sequential<Convolution2D<E, D>, BatchNorm<E, D>>, Sequential<Relu<E, D>, MaxPool2D<E, D>>>

Swift

public var conv2: Sequential<Sequential<Convolution2D<E, D>, BatchNorm<E, D>>, Sequential<Relu<E, D>, MaxPool2D<E, D>>>

Swift

public var conv3: Sequential<Sequential<Sequential<Convolution2D<E, D>, BatchNorm<E, D>>, Sequential<Relu<E, D>, Convolution2D<E, D>>>, Sequential<Sequential<BatchNorm<E, D>, Relu<E, D>>, MaxPool2D<E, D>>>

Swift

public var conv4: Sequential<Sequential<Sequential<Convolution2D<E, D>, BatchNorm<E, D>>, Sequential<Relu<E, D>, Convolution2D<E, D>>>, Sequential<Sequential<BatchNorm<E, D>, Relu<E, D>>, MaxPool2D<E, D>>>

Swift

public var conv5: Sequential<Sequential<Sequential<Convolution2D<E, D>, BatchNorm<E, D>>, Sequential<Relu<E, D>, Convolution2D<E, D>>>, Sequential<Sequential<BatchNorm<E, D>, Relu<E, D>>, MaxPool2D<E, D>>>

Swift

public var dense: DenseLayer

Swift

public init(inputChannels: Int, classes: Int)