首先这个weight参数比想象中的要考虑的多,你可以试试下面代码import ... 关于pytorch的CrossEntropyLoss的weight参数 ... CLASS torch.nn.
確定! 回上一頁