pytorch中交叉熵函数torch.nn.CrossEntropyLoss()怎么加入权重
关于交叉熵函数torch.nn.CrossEntropyLoss的基本用法可参考https://blog.csdn.net/zziahgf/article/details/80196376。那么如何在交叉熵函数中添加自定义的各类别的权重呢?定义参数weight即可,其中参数的内容要为tensor类型,而且要把dtype类型定义为float。用法为:### 我的类别数为12criterion ...
原创
2019-07-30 15:25:54 ·
16756 阅读 ·
12 评论