Class MnistLeNet

Inheritance Relationships

Base Type

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