costume loss:
def loss_function(p, y, b):
losses = 0
for i in range(b):
k = len(y[i])
loss1 = -(torch.ones(k).cuda()-y[i])@((torch.ones(k).cuda()-p[i]).log())
l = y[i]*(p[i].log())
prod = (torch.ones(k).cuda()-y[i]) - l
loss2 = loss1 + torch.max(torch.ones(k).cuda()-prod)
losses = loss1 + loss2
return Variable(losses/(b))
dont know how to correct this error