关于pytorch在使用torch.nn.functional.cross_entropy的注意点

在使用pytorch实现textcnn进行文本分类时,参考了别人的代码(keras),在使用torch.nn.functional.cross_entropy纠结了好久,最后发现的下面的文章:html https://www.cnblogs.com/marsggbo/p/10401215.html网络 总结了如下的注意点:编码 在网络的forward最后输出时不用接softmax,直接全链接输出n
相关文章
相关标签/搜索