Train in own dataset

[aichifandefan]

How to train ous dataset?
when i train in my dataset it get

pos_loss = torch.log(pred) * torch.pow(1 - pred, 2) * pos_inds
RuntimeError: The size of tensor a (80) must match the size of tensor b (2) at non-singleton dimension 1

Is the classes number problem?
I can not find the classes config

[aichifandefan]

My dataset have 2 classes, blackground and book.

[aichifandefan]

Yeah I find de classes parameters
model.heads['ct_hm'] = 80
when i change it to 2 it can run