A question about pred and target in ghm_loss

[DangerousY]

The method is very good! However, I am new in this area. So I see the ghm code include two parameter: pred [batch_num, class_num] and target [batch_num, class_num]. I want to ask if the pred only input the [batch,class] matrix, we don't need to input the image's C x H x W. Because I see that the input of focal_loss is pred[batch, c, h, w], I am confused.
Tks for your reply.