nebula.core.models.cifar10.cnn#
Classes#
Abstract class for the NEBULA model. |
Module Contents#
- class nebula.core.models.cifar10.cnn.CIFAR10ModelCNN(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#
- conv2#
- conv3#
- pool#
- fc1#
- fc2#
- forward(x)#
Forward pass of the model.
- configure_optimizers()#
Optimizer configuration.