The multi-line comment format we use is:
/**
 * NOTE(xinghao): 
 */

Or in this case, you could just use a single line comment format

// NOTE: The loss here ...
// and reg_val