Coarse training

Fit

void eddl::fit(model m, const vector<Tensor*> &in, const vector<Tensor*> &out, int batch, int epochs)

Trains the model for a fixed number of epochs (iterations on a dataset).

Parameters
  • m – Model to train

  • in – Input data (features)

  • out – Output data (labels)

  • batch – Number of samples per gradient update

  • epochs – Number of epochs to train the model. An epoch is an iteration over the entire data provided

Returns

(void) Trains the model

Example:

model net = Model({in}, {out});

// Build model
...

// Train model
fit(net, {x_train}, {y_train}, batch_size, epochs);