nebula.core.models.cifar10.cnnV2

nebula.core.models.cifar10.cnnV2#

Classes#

CIFAR10ModelCNN_V2

Abstract class for the NEBULA model.

Module Contents#

class nebula.core.models.cifar10.cnnV2.CIFAR10ModelCNN_V2(input_channels=3, num_classes=10, learning_rate=0.001, metrics=None, confusion_matrix=None, seed=None)#

Bases: nebula.core.models.nebulamodel.NebulaModel

Abstract class for the NEBULA model.

This class is an abstract class that defines the interface for the NEBULA model.

config#
example_input_array#
learning_rate#
criterion#
conv1#
bn1#
conv2#
bn2#
conv3#
bn3#
pool#
fc1#
fc2#
dropout#
forward(x)#

Forward pass of the model.

configure_optimizers()#

Optimizer configuration.

Contents