small error in class_balanced_loss.py

wgqtmac / cvprw2020

The code for Chalearn Single-modal (RGB) Cross-ethnicity Face anti-spoofing Recognition Challenge@CVPR2020

46 stars 7 forks source link

small error in class_balanced_loss.py #4

Open eeric opened 4 years ago

eeric commented 4 years ago

on 87th line in class_balanced_loss.py: cb_loss = F.binary_cross_entropy_with_logits(input = logits,target = labels_one_hot, weights = weights) there was small error that 'weights = weights' should be modified for 'weight = weights'.

wgqtmac commented 4 years ago

Thanks for your attention! We will change this error.