Line 87 typo

[BenDrewry]

trafficstars

original: cb_loss = F.binary_cross_entropy_with_logits(input = logits,target = labels_one_hot, weights = weights)

update: cb_loss = F.binary_cross_entropy_with_logits(input = logits,target = labels_one_hot, weight = weights)

• pytorch optimizer takes keyword 'weight', not 'weights'

Thanks for the resource, it worked great!!

[Python-Eric]

wo ! useful , it did work !