Open aravindhv10 opened 2 months ago
when training with batch size > 1, there is a broadcasting issue. This fixes it.
when training with batch size > 1, there is a broadcasting issue. This fixes it.