火炬损失
我实现的标签平滑,amsoftmax,焦点损耗,双焦点损耗,三重态损耗,giou损耗,亲和力损耗,pc_softmax_cross_entropy,ohem损耗(基于行硬挖掘损失的softmax),大利润- softmax(bmvc2019),lovasz-softmax-loss和dice-loss(广义的软骰子损失和批处理软骰子损失)。 也许这对我的未来工作很有用。
还尝试实现swish,hard-swish(hswish)和mish激活功能。
此外,添加了基于cuda的一键式功能(支持标签平滑)。
新添加一个“指数移动平均线(EMA)”运算符。
添加卷积运算,例如coord-conv2d和dynamic-conv2d(dy-conv2d)。
一些运算符是使用pytorch cuda扩展实现的,因此您需要先对其进行编译:
$ python setup.py
1