[legacy-nn] losses need to return tensors and not numbers #83
Description
right now, losses return numbers.
This is a sync point and we should avoid this earlier in the scheme of things than push it in later.
{{ message }}
right now, losses return numbers.
This is a sync point and we should avoid this earlier in the scheme of things than push it in later.