setTrainingMode function

void setTrainingMode(
  1. SNetwork model,
  2. bool isTraining
)

Iterates through a model's layers and sets the isTraining flag on any DropoutLayer instances.

Implementation

void setTrainingMode(SNetwork model, bool isTraining) {
  for (Layer layer in model.layers) {
    if (layer is DropoutLayer) {
      layer.isTraining = isTraining;
    }
  }
}