Lines Matching defs:param
216 fun update_layer param (layer, layerwu) =
218 val coeff = #learning_rate param / Real.fromInt (#batch_size param)
226 fun update_nn param nn wu = map (update_layer param) (combine (nn,wu))
247 fun train_nn_batch param pf nn batch =
252 val newnn = update_nn param nn dwl
257 fun train_nn_epoch param pf lossl nn batchl = case batchl of
260 let val (newnn,loss) = train_nn_batch param pf nn batch in
261 train_nn_epoch param pf (loss :: lossl) newnn m
264 fun train_nn_nepoch param pf i nn exl =
265 if i >= #nepoch param then nn else
267 val batchl = mk_batch (#batch_size param) (shuffle exl)
268 val (new_nn,loss) = train_nn_epoch param pf [] nn batchl
270 if #verbose param then print_endline (its i ^ " " ^ sr loss) else ()
272 train_nn_nepoch param pf (i+1) new_nn exl
288 fun train_nn param nn exl =
290 val (pf,close_threadl) = parmap_gen (#ncore param)
291 val _ = if #verbose param then stats_exl exl else ()
293 val r = train_nn_nepoch param pf 0 nn newexl
322 val param : trainparam =
326 val (newnn,t) = add_time (train_nn param nn) exl;