AdamW (identical to torch. pytorch交叉熵损失函数的weight参数的使用首先必须将权重也转为Tensor的cuda格式: 然后将该class_weight作为交叉熵函数对应参数的输入值.
確定! 回上一頁