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
How to train ous dataset? when i train in my dataset it get
Is the classes number problem? I can not find the classes config