nebula.core.models.cifar10.simplemobilenet

Classes

SimpleMobileNetV1

Module Contents

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

Bases: lightning.LightningModule

process_metrics(phase, y_pred, y, loss=None)

log_metrics_by_epoch(phase, print_cm=False, plot_cm=False)

config

example_input_array

learning_rate

criterion

model

fc

forward(x)

configure_optimizers()