Fix bug in WeightedTVLoss when `weight` is `None` (#420)
* Fix an error when `weight` is `None` * Update losses.py * Update losses.py Co-authored-by: Xintao <wxt1994@126.com> Co-authored-by: Xintao <xintao.alpha@gmail.com>
W
wwhio committed
29d79f238c416eb9be0a6868491b4fb54de9ad41
Parent: fcf9489
Committed by GitHub <noreply@github.com>
on 8/29/2021, 2:27:59 PM