train

fun train()(source)

Set the module to training mode