maybe inf and nans when class not present in batch at all. maybe train error huge because forgot to divide loss by num
fixed convlict