CNN

class fortuna.model.lenet.LeNet5(output_dim, dtype=<class 'jax.numpy.float32'>, parent=<flax.linen.module._Sentinel object>, name=None)

Bases: Module

A LeNet-5 network [LeCun et al., 1989](http://yann.lecun.com/exdb/publis/pdf/lecun-89e.pdf). Please refer to Model for the internal methods.

output_dim

The output model dimension.

Type:

int

dtype

Layers’ dtype.

Type:

Any