Class MnistLeNet

• Defined in File model_lenet.h

Inheritance Relationships

Base Type

• public marian::models::MnistFeedForwardNet (Class MnistFeedForwardNet)

Class Documentation

class MnistLeNet : public marian::models::MnistFeedForwardNet

Public Functions

template<class ...Args>
MnistLeNet(Ptr<Options> options, Args... args)

virtual void clear(Ptr<ExpressionGraph> graph)

Protected Functions

virtual Expr apply(Ptr<ExpressionGraph> g, Ptr<data::Batch> batch, bool = false)

Builds an expression graph representing a feed-forward classifier.

Return

a shared pointer to the newly constructed expression graph

Parameters

• dims: number of nodes in each layer of the feed-forward classifier

• batch: a batch of training or testing examples

• training: create a classifier for training or for inference only