ahrefs / ocannl

OCANNL: OCaml Compiles Algorithms for Neural Networks Learning
BSD 2-Clause "Simplified" License
67 stars 2 forks source link

Make name passing optional by: extracting names from the comments in `Assignments.t`, generating more comments #198

Closed lukstafi closed 1 year ago

lukstafi commented 1 year ago

Having to provide names is annoying since there should be canonical names for code derived for specific purposes. E.g. for Train.update_step we could have label_fwd_bprop, or label_update_step, where "label" is the label of the tensor; for Train.sgd_step we could have label_sgd_step as the name.