std :: ml::module
Function std::module_fit
generic<M: Module> struct TrainReport module_fit(
&M m,
int loss,
&Optimizer opt,
const &Dataset data,
const &TrainConfig cfg,
)trainer_fit with a built-in loss: the same batches, steps, lr decay and report. opt holds m's parameters (model_add_params).